Hello_User/gpu-jupyter
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

closes #25 by adding libcudnn7-dev

Browse Source
This commit is contained in:
Michael Pilosov 2020-08-31 21:23:02 -06:00 committed by GitHub
parent 1c8b32a23c
commit 113d082cc4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/Dockerfile.gpulibs
View File

@ -15,6 +15,9 @@ RUN pip install --upgrade pip && \
RUN conda install --quiet --yes \ RUN conda install --quiet --yes \
pyyaml mkl mkl-include setuptools cmake cffi typing 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/ # Check compatibility here: https://pytorch.org/get-started/locally/
RUN conda install --quiet --yes \ RUN conda install --quiet --yes \
pytorch \ pytorch \