Skip to content

Commit 0b937a9

Browse files
Qi Qiaoicywang86rui
authored andcommitted
Update CuDNN and NCCL2 versions. (#214)
1 parent 8fd4b49 commit 0b937a9

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

docker/1.13.1/Dockerfile.gpu

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -12,11 +12,11 @@ RUN apt-get update && apt-get install -y --no-install-recommends --allow-unauthe
1212
cuda-cusolver-dev-10-0 \
1313
cuda-cusparse-dev-10-0 \
1414
curl \
15-
libcudnn7=7.4.1.5-1+cuda10.0 \
15+
libcudnn7=7.5.1.10-1+cuda10.0 \
1616
# TensorFlow doesn't require libnccl anymore but Open MPI still depends on it
17-
libnccl2=2.3.5-2+cuda10.0 \
17+
libnccl2=2.4.7-1+cuda10.0 \
1818
libgomp1 \
19-
libnccl-dev=2.3.5-2+cuda10.0 \
19+
libnccl-dev=2.4.7-1+cuda10.0 \
2020
libfreetype6-dev \
2121
libhdf5-serial-dev \
2222
libpng12-dev \

0 commit comments

Comments
 (0)