experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Upgrade ci docker images to cuda 10.

Browse Source 
These really should be gone already...

PiperOrigin-RevId: 227914258
This commit is contained in:
Gunhan Gulsoy 2019-01-04 14:27:22 -08:00 committed by TensorFlower Gardener
parent dba2134638
commit 2f246ef177
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/tools/ci_build/Dockerfile.gpu
View File

@ -1,4 +1,4 @@
FROM nvidia/cuda:9.0-cudnn7-devel-ubuntu16.04
FROM nvidia/cuda:10.0-cudnn7-devel-ubuntu16.04
LABEL maintainer="Jan Prach <jendap@google.com>"
@ -24,7 +24,7 @@ COPY install/.bazelrc /etc/bazel.bazelrc
ENV LD_LIBRARY_PATH /usr/local/cuda/extras/CUPTI/lib64:$LD_LIBRARY_PATH
# Link NCCL libray and header where the build script expects them.
RUN mkdir /usr/local/cuda-9.0/lib && \
RUN mkdir /usr/local/cuda/lib && \
ln -s /usr/lib/x86_64-linux-gnu/libnccl.so.2 /usr/local/cuda/lib/libnccl.so.2 && \
ln -s /usr/include/nccl.h /usr/local/cuda/include/nccl.h

2
tensorflow/tools/ci_build/Dockerfile.rbe.gpu
View File

@ -1,4 +1,4 @@
FROM nvidia/cuda:9.0-cudnn7-devel-ubuntu16.04
FROM nvidia/cuda:10.0-cudnn7-devel-ubuntu16.04
LABEL maintainer="Nick Lopez <ngiraldo@google.com>"