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

Merge pull request #26 from mathematicalmichael/patch-2

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