This is an automated email from the ASF dual-hosted git repository.
zhasheng pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git
The following commit(s) were added to refs/heads/master by this push:
new e5f0c52 CI: switch to cudnn8 devel docker as it is now available
(#19696)
e5f0c52 is described below
commit e5f0c524f141568b8548e2870a48daa129800036
Author: Leonard Lausen <[email protected]>
AuthorDate: Sat Dec 19 18:03:03 2020 -0600
CI: switch to cudnn8 devel docker as it is now available (#19696)
https://gitlab.com/nvidia/container-images/cuda/-/issues/83#note_469420005
---
ci/docker/Dockerfile.build.ubuntu | 4 +---
ci/docker/docker-compose.yml | 4 ++--
2 files changed, 3 insertions(+), 5 deletions(-)
diff --git a/ci/docker/Dockerfile.build.ubuntu
b/ci/docker/Dockerfile.build.ubuntu
index 952f4c4..95a322d 100644
--- a/ci/docker/Dockerfile.build.ubuntu
+++ b/ci/docker/Dockerfile.build.ubuntu
@@ -144,7 +144,7 @@ COPY runtime_functions.sh /work/
####################################################################################################
FROM base as gpu
-# Install TensorRT and CuDNN
+# Install TensorRT
# Use bash as it has better support for string comparisons in if clauses
SHELL ["/bin/bash", "-c"]
# We need to redeclare ARG due to
@@ -169,7 +169,5 @@ RUN export SHORT_CUDA_VERSION=${CUDA_VERSION%.*} && \
libnvinfer-plugin${TRT_MAJOR_VERSION}=${TRT_VERSION} \
libnvinfer-plugin-dev=${TRT_VERSION}; \
fi && \
- apt-get install -y libcudnn8-dev && \
rm -rf /var/lib/apt/lists/*
-ENV CUDNN_VERSION=8.0.5
diff --git a/ci/docker/docker-compose.yml b/ci/docker/docker-compose.yml
index 6ebc6a0..7c0ef43 100644
--- a/ci/docker/docker-compose.yml
+++ b/ci/docker/docker-compose.yml
@@ -95,7 +95,7 @@ services:
dockerfile: Dockerfile.build.ubuntu
target: gpu
args:
- BASE_IMAGE: nvidia/cuda:11.1-devel-ubuntu18.04
+ BASE_IMAGE: nvidia/cuda:11.1-cudnn8-devel-ubuntu18.04
cache_from:
- ${DOCKER_CACHE_REGISTRY}/build.ubuntu_tensorrt_cu111:latest
ubuntu_gpu_cu111:
@@ -105,7 +105,7 @@ services:
dockerfile: Dockerfile.build.ubuntu
target: gpu
args:
- BASE_IMAGE: nvidia/cuda:11.1-devel-ubuntu20.04
+ BASE_IMAGE: nvidia/cuda:11.1-cudnn8-devel-ubuntu20.04
cache_from:
- ${DOCKER_CACHE_REGISTRY}/build.ubuntu_gpu_cu111:latest
###################################################################################################
