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

back to cuda 10.1

Browse Source
This commit is contained in:
Christoph Schranz 2020-02-23 20:26:12 +01:00
parent 7d015b6da3
commit ba5b99e91c
7 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
.build/Dockerfile Normal file → Executable file
View File

@ -1,6 +1,6 @@
# 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
FROM nvidia/cuda:10.2-base-ubuntu18.04
FROM nvidia/cuda:10.1-base-ubuntu18.04
LABEL maintainer="Jupyter Project <jupyter@googlegroups.com>"
############################################################################
@ -380,7 +380,7 @@ RUN conda install --quiet --yes \
# Install PyTorch, version of cudatoolkit must match those of the base image
RUN conda install -y -c pytorch \
cudatoolkit=10.2 \
cudatoolkit=10.1 \
'pytorch=1.3.1' \
torchvision && \
conda clean --all -f -y && \

0
.build/fix-permissions Normal file → Executable file
View File

0
.build/jupyter_notebook_config.json Normal file → Executable file
View File

0
.build/jupyter_notebook_config.py Normal file → Executable file
View File

2
generate_Dockerfile.sh
View File

@ -29,7 +29,7 @@ cp $STACKS_DIR/base-notebook/fix-permissions .build/
cp $STACKS_DIR/base-notebook/start.sh .build/
cp $STACKS_DIR/base-notebook/start-notebook.sh .build/
cp $STACKS_DIR/base-notebook/start-singleuser.sh .build/
chmod 755 .build/*.sh
chmod 755 .build/*
echo "
############################################################################

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.
# 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>"

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
RUN conda install -y -c pytorch \
cudatoolkit=10.2 \
cudatoolkit=10.1 \
'pytorch=1.3.1' \
torchvision && \
conda clean --all -f -y && \