use nvidia docker and install latest cuDNN

apache · May 23, 2019 · aff1fdf · aff1fdf
1 parent 6a90f58
commit aff1fdf
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/ci/docker/Dockerfile.publish.ubuntu1404_gpu b/ci/docker/Dockerfile.publish.ubuntu1404_gpu
@@ -18,13 +18,17 @@
 #
 # Dockerfile to run MXNet on Ubuntu 14.04 for GPU
 
-FROM ubuntu:14.04
+FROM nvidia/cuda:10.1-cudnn7-devel-ubuntu16.04
 
 WORKDIR /work/deps
 
 COPY install/ubuntu_publish.sh /work/
 RUN /work/ubuntu_publish.sh
 
+ENV CUDNN_VERSION=7.5.1.10
+COPY install/ubuntu_cudnn.sh /work/
+RUN /work/ubuntu_cudnn.sh
+
 ARG USER_ID=0
 ARG GROUP_ID=0
 COPY install/ubuntu_adduser.sh /work/