closes #25 by adding libcudnn7-dev

2020-08-31 21:23:02 -06:00 · 2020-08-31 21:23:02 -06:00 · 113d082cc4
commit 113d082cc4
parent 1c8b32a23c
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 \