diff options
author | Felix Abecassis <felix.abecassis@gmail.com> | 2018-06-01 04:45:15 +0200 |
---|---|---|
committer | Gunhan Gulsoy <gunan@google.com> | 2018-05-31 19:45:15 -0700 |
commit | ae3456402ca15309a2fcb85adbaa8b464ca2d065 (patch) | |
tree | 62a8a29d889c2446bd20f5af68d6f82e098e1b7a /tensorflow/tools/docker | |
parent | 19ab879e55e7e41923f7999d2f12793d849b24d0 (diff) |
docker: update cuDNN to 7.1.4.18 (#19636)
Signed-off-by: Felix Abecassis <fabecassis@nvidia.com>
Diffstat (limited to 'tensorflow/tools/docker')
-rw-r--r-- | tensorflow/tools/docker/Dockerfile.devel-gpu | 4 | ||||
-rw-r--r-- | tensorflow/tools/docker/Dockerfile.gpu | 2 |
2 files changed, 3 insertions, 3 deletions
diff --git a/tensorflow/tools/docker/Dockerfile.devel-gpu b/tensorflow/tools/docker/Dockerfile.devel-gpu index 2fe47f3356..e4dcce9cdd 100644 --- a/tensorflow/tools/docker/Dockerfile.devel-gpu +++ b/tensorflow/tools/docker/Dockerfile.devel-gpu @@ -13,8 +13,8 @@ RUN apt-get update && apt-get install -y --no-install-recommends \ cuda-cusparse-dev-9-0 \ curl \ git \ - libcudnn7=7.0.5.15-1+cuda9.0 \ - libcudnn7-dev=7.0.5.15-1+cuda9.0 \ + libcudnn7=7.1.4.18-1+cuda9.0 \ + libcudnn7-dev=7.1.4.18-1+cuda9.0 \ libcurl3-dev \ libfreetype6-dev \ libhdf5-serial-dev \ diff --git a/tensorflow/tools/docker/Dockerfile.gpu b/tensorflow/tools/docker/Dockerfile.gpu index bff4a20392..9197651ff4 100644 --- a/tensorflow/tools/docker/Dockerfile.gpu +++ b/tensorflow/tools/docker/Dockerfile.gpu @@ -12,7 +12,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \ cuda-cusolver-9-0 \ cuda-cusparse-9-0 \ curl \ - libcudnn7=7.0.5.15-1+cuda9.0 \ + libcudnn7=7.1.4.18-1+cuda9.0 \ libfreetype6-dev \ libhdf5-serial-dev \ libpng12-dev \ |