From 8a6c83656a2197309dacba124944c665530dd218 Mon Sep 17 00:00:00 2001 From: "William D. Irons" Date: Thu, 13 Sep 2018 11:19:41 -0400 Subject: Move ppc64le build/test to cuda 9.2 CUDA 9.2 is the first version of cuda that nvidia will support on Power9 hardware. The dockerfile is used in the jenkins build of the pp64le whl file and in the CI/CD test. --- tensorflow/tools/ci_build/Dockerfile.gpu.ppc64le | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'tensorflow/tools/ci_build') diff --git a/tensorflow/tools/ci_build/Dockerfile.gpu.ppc64le b/tensorflow/tools/ci_build/Dockerfile.gpu.ppc64le index e026edb6bb..0a55b84ac4 100644 --- a/tensorflow/tools/ci_build/Dockerfile.gpu.ppc64le +++ b/tensorflow/tools/ci_build/Dockerfile.gpu.ppc64le @@ -1,4 +1,4 @@ -FROM nvidia/cuda-ppc64le:9.0-cudnn7-devel-ubuntu16.04 +FROM nvidia/cuda-ppc64le:9.2-cudnn7-devel-ubuntu16.04 LABEL maintainer="William Irons " @@ -26,6 +26,8 @@ ENV LD_LIBRARY_PATH /usr/local/cuda/extras/CUPTI/lib64:$LD_LIBRARY_PATH # Configure the build for our CUDA configuration. ENV TF_NEED_CUDA 1 ENV TF_CUDA_COMPUTE_CAPABILITIES 3.0 +ENV TF_CUDA_VERSION 9.2 +ENV CUDA_TOOLKIT_PATH /usr/local/cuda-9.2 # TODO get NCCL 2 in the docker image ENV TF_NCCL_VERSION 1 -- cgit v1.2.3