diff --git a/ci/docker/Dockerfile.build.ubuntu_gpu_tensorrt b/ci/docker/Dockerfile.build.ubuntu_gpu_tensorrt index 8ad90aedeb6f..09591ee0b9c9 100644 --- a/ci/docker/Dockerfile.build.ubuntu_gpu_tensorrt +++ b/ci/docker/Dockerfile.build.ubuntu_gpu_tensorrt @@ -18,7 +18,7 @@ # # Dockerfile to run MXNet on Ubuntu 16.04 for CPU -FROM nvidia/cuda:10.0-cudnn7-devel +FROM nvidia/cuda:10.0-devel WORKDIR /work/deps @@ -35,6 +35,10 @@ ARG USER_ID=0 COPY install/ubuntu_adduser.sh /work/ RUN /work/ubuntu_adduser.sh +ENV CUDNN_VERSION=7.5.0.56 +COPY install/ubuntu_cudnn.sh /work/ +RUN /work/ubuntu_cudnn.sh + COPY runtime_functions.sh /work/ WORKDIR /work/mxnet diff --git a/ci/docker/install/tensorrt.sh b/ci/docker/install/tensorrt.sh index 1950cad0b52f..21d9b4d40b36 100755 --- a/ci/docker/install/tensorrt.sh +++ b/ci/docker/install/tensorrt.sh @@ -44,5 +44,6 @@ echo "TensorRT build enabled. Installing TensorRT." wget -qO tensorrt.deb https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1604/x86_64/nvinfer-runtime-trt-repo-ubuntu1604-5.0.2-ga-cuda10.0_1-1_amd64.deb dpkg -i tensorrt.deb apt-get update -apt-get install -y --allow-downgrades libnvinfer-dev +apt-get install -y --allow-downgrades libnvinfer5=5.1.2-1+cuda10.0 +apt-get install -y --allow-downgrades libnvinfer-dev=5.1.2-1+cuda10.0 rm tensorrt.deb diff --git a/ci/docker/install/ubuntu_cudnn.sh b/ci/docker/install/ubuntu_cudnn.sh index 3f649074f4d5..3d260046b5e7 100755 --- a/ci/docker/install/ubuntu_cudnn.sh +++ b/ci/docker/install/ubuntu_cudnn.sh @@ -50,5 +50,5 @@ case ${CUDA_VERSION} in ;; esac -apt-get install -y libcudnn7=${libcudnn7_version} libcudnn7-dev=${libcudnn7_dev_version} +apt-get install -y --allow-downgrades libcudnn7=${libcudnn7_version} libcudnn7-dev=${libcudnn7_dev_version} diff --git a/ci/jenkins/Jenkinsfile_unix_gpu b/ci/jenkins/Jenkinsfile_unix_gpu index 886842712879..e2a089b0469b 100644 --- a/ci/jenkins/Jenkinsfile_unix_gpu +++ b/ci/jenkins/Jenkinsfile_unix_gpu @@ -39,7 +39,7 @@ core_logic: { custom_steps.compile_unix_full_gpu(), custom_steps.compile_unix_cmake_mkldnn_gpu(), custom_steps.compile_unix_cmake_gpu(), - // custom_steps.compile_unix_tensorrt_gpu(), + custom_steps.compile_unix_tensorrt_gpu(), custom_steps.compile_unix_int64_gpu() ]) @@ -51,7 +51,7 @@ core_logic: { custom_steps.test_unix_python2_mkldnn_gpu(), custom_steps.test_unix_python3_mkldnn_gpu(), custom_steps.test_unix_python3_mkldnn_nocudnn_gpu(), - // custom_steps.test_unix_python3_tensorrt_gpu(), + custom_steps.test_unix_python3_tensorrt_gpu(), custom_steps.test_unix_perl_gpu(), custom_steps.test_unix_r_gpu(), custom_steps.test_unix_cpp_gpu(),