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

rebase to nvidia/cuda:10.1-base-ubuntu18.04 as 10.2 makes problems

Browse Source
This commit is contained in:
Christoph Schranz
2020-02-22 20:50:43 +01:00
parent 6a0dd5607e
commit 5b54ba148c
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/Dockerfile.header
View File

@@ -1,4 +1,4 @@
# Use NVIDIA CUDA as base image and run the same installation as in the other packages. # Use NVIDIA CUDA as base image and run the same installation as in the other packages.
# The version of cudatoolkit must match those of the base image, see Dockerfile.pytorch # The version of cudatoolkit must match those of the base image, see Dockerfile.pytorch
FROM nvidia/cuda:10.2-base-ubuntu18.04 FROM nvidia/cuda:10.1-base-ubuntu18.04
LABEL maintainer="Jupyter Project <jupyter@googlegroups.com>" LABEL maintainer="Jupyter Project <jupyter@googlegroups.com>"

2
src/Dockerfile.pytorch
View File

@@ -7,7 +7,7 @@ RUN conda install --quiet --yes \
# Install PyTorch, version of cudatoolkit must match those of the base image # Install PyTorch, version of cudatoolkit must match those of the base image
RUN conda install -y -c pytorch \ RUN conda install -y -c pytorch \
cudatoolkit=10.2 \ cudatoolkit=10.1 \
'pytorch=1.3.1' \ 'pytorch=1.3.1' \
torchvision && \ torchvision && \
conda clean --all -f -y && \ conda clean --all -f -y && \