diff options
author | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-15 13:14:20 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-15 13:14:29 -0700 |
commit | e37baff90ed84faef679bbf1e785e6790fd2739d (patch) | |
tree | e9e659317534168589cf4080b4a52a67551b32f3 /tensorflow/tools/ci_build | |
parent | eab14a9303f6268d97fa3b901cc09a71c86bba63 (diff) | |
parent | 8a6c83656a2197309dacba124944c665530dd218 (diff) |
Merge pull request #22255 from wdirons:ppc64le_use_cuda92
PiperOrigin-RevId: 213128841
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 |