Merge pull request #26 from mathematicalmichael/patch-2

closes #25 by adding libcudnn7-dev Thank you all!
iot-salzburg · Sep 1, 2020 · 73d4188 · 73d4188
2 parents 1c8b32a + 113d082
commit 73d4188
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/Dockerfile.gpulibs b/src/Dockerfile.gpulibs
@@ -15,6 +15,9 @@ RUN pip install --upgrade pip && \
 RUN conda install --quiet --yes \
     pyyaml mkl mkl-include setuptools cmake cffi typing
 
+# Deep-NN libs
+RUN apt-get update && apt-get install -y libcudnn7-dev && rm -rf /var/lib/apt/lists/* 
+
 # Check compatibility here: https://pytorch.org/get-started/locally/
 RUN conda install --quiet --yes \
      pytorch \