diff options
Diffstat (limited to 'tensorflow/tools/ci_build/Dockerfile.gpu')
| -rw-r--r-- | tensorflow/tools/ci_build/Dockerfile.gpu | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/tools/ci_build/Dockerfile.gpu b/tensorflow/tools/ci_build/Dockerfile.gpu index 7b09691178..b36edb9bde 100644 --- a/tensorflow/tools/ci_build/Dockerfile.gpu +++ b/tensorflow/tools/ci_build/Dockerfile.gpu @@ -22,5 +22,6 @@ ENV LD_LIBRARY_PATH /usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64 # Configure the build for our CUDA configuration. ENV CUDA_TOOLKIT_PATH /usr/local/cuda -ENV CUDNN_INSTALL_PATH /usr/local/cuda +ENV CUDNN_INSTALL_PATH /usr/lib/x86_64-linux-gnu ENV TF_NEED_CUDA 1 +ENV CUDA_COMPUTE_CAPABILITIES 3.0,5.2 |