Use the tarball version of cudnn since tvm has trouble detecting the …

…other one
apache · Jun 13, 2019 · 335dbf9 · 335dbf9
1 parent efa0290
commit 335dbf9
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 9 deletions.
diff --git a/conda/Dockerfile.template b/conda/Dockerfile.template
@@ -18,11 +18,15 @@
 FROM nvidia/cuda:{{ cuda_version }}-devel-ubuntu16.04
 
 RUN apt-get update && apt-get install -y --no-install-recommends \
-            bzip2 curl sudo binutils \
-            libcudnn7={{ cudnn_version }}-1+cuda{{ cuda_version }} \
-            libcudnn7-dev={{ cudnn_version }}-1+cuda{{ cuda_version }} && \
+            bzip2 curl sudo binutils && \
     rm -rf /var/lib/apt/lists/*
 
+RUN  curl -fsSL http://developer.download.nvidia.com/compute/redist/cudnn/v{{ cudnn_short_version }}/cudnn-{{ cuda_version }}-linux-x64-v{{ cudnn_version }}.tgz -O && \
+    tar --no-same-owner -xzf cudnn-{{ cuda_version }}-linux-x64-v{{ cudnn_version }}.tgz -C /usr/local && \
+    rm cudnn-{{ cuda_version }}-linux-x64-v{{ cudnn_version }}.tgz && \
+    ldconfig
+
+
 RUN curl -o ~/miniconda.sh -O https://repo.continuum.io/miniconda/Miniconda3-latest-Linux-x86_64.sh  && \
      chmod +x ~/miniconda.sh && \
      ~/miniconda.sh -b -p /opt/conda && \

diff --git a/docker/Dockerfile.conda_cuda100 b/docker/Dockerfile.conda_cuda100
@@ -18,11 +18,15 @@
 FROM nvidia/cuda:10.0-devel-ubuntu16.04
 
 RUN apt-get update && apt-get install -y --no-install-recommends \
-            bzip2 curl sudo binutils \
-            libcudnn7=7.6.0.64-1+cuda10.0 \
-            libcudnn7-dev=7.6.0.64-1+cuda10.0 && \
+            bzip2 curl sudo binutils && \
     rm -rf /var/lib/apt/lists/*
 
+RUN  curl -fsSL http://developer.download.nvidia.com/compute/redist/cudnn/v7.6.0/cudnn-10.0-linux-x64-v7.6.0.64.tgz -O && \
+    tar --no-same-owner -xzf cudnn-10.0-linux-x64-v7.6.0.64.tgz -C /usr/local && \
+    rm cudnn-10.0-linux-x64-v7.6.0.64.tgz && \
+    ldconfig
+
+
 RUN curl -o ~/miniconda.sh -O https://repo.continuum.io/miniconda/Miniconda3-latest-Linux-x86_64.sh  && \
      chmod +x ~/miniconda.sh && \
      ~/miniconda.sh -b -p /opt/conda && \

diff --git a/docker/Dockerfile.conda_cuda90 b/docker/Dockerfile.conda_cuda90
@@ -18,11 +18,15 @@
 FROM nvidia/cuda:9.0-devel-ubuntu16.04
 
 RUN apt-get update && apt-get install -y --no-install-recommends \
-            bzip2 curl sudo binutils \
-            libcudnn7=7.6.0.64-1+cuda9.0 \
-            libcudnn7-dev=7.6.0.64-1+cuda9.0 && \
+            bzip2 curl sudo binutils && \
     rm -rf /var/lib/apt/lists/*
 
+RUN  curl -fsSL http://developer.download.nvidia.com/compute/redist/cudnn/v7.6.0/cudnn-9.0-linux-x64-v7.6.0.64.tgz -O && \
+    tar --no-same-owner -xzf cudnn-9.0-linux-x64-v7.6.0.64.tgz -C /usr/local && \
+    rm cudnn-9.0-linux-x64-v7.6.0.64.tgz && \
+    ldconfig
+
+
 RUN curl -o ~/miniconda.sh -O https://repo.continuum.io/miniconda/Miniconda3-latest-Linux-x86_64.sh  && \
      chmod +x ~/miniconda.sh && \
      ~/miniconda.sh -b -p /opt/conda && \