closes #25 by adding libcudnn7-dev

iot-salzburg · Sep 1, 2020 · 113d082 · 113d082
1 parent 1c8b32a
commit 113d082
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 \