diff options
author | 2018-09-13 11:19:41 -0400 | |
---|---|---|
committer | 2018-09-13 11:19:41 -0400 | |
commit | 8a6c83656a2197309dacba124944c665530dd218 (patch) | |
tree | e76014d781234719c764b06da1230a19f873b36b /tensorflow/tools/ci_build | |
parent | 9c270922715306efefce848b87dee3690cdddd27 (diff) |
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.
Diffstat (limited to 'tensorflow/tools/ci_build')
-rw-r--r-- | tensorflow/tools/ci_build/Dockerfile.gpu.ppc64le | 4 |
1 files changed, 3 insertions, 1 deletions
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 <wdirons@us.ibm.com>" @@ -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 |