Merge pull request #26 from mathematicalmichael/patch-2

closes #25 by adding libcudnn7-dev Thank you all!
2020-09-01 08:23:34 +02:00
parent 1c8b32a23c 113d082cc4
commit 73d4188b5a
1 changed files with 3 additions and 0 deletions
--- 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 \