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

build image in .build/

Browse Source 
nvidia/cuda:10.2-base-ubuntu18.04
This commit is contained in:
Christoph Schranz
2020-02-22 20:40:18 +01:00
parent 4f48080c69
commit 6a0dd5607e
10 changed files with 7 additions and 749 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
.build/Dockerfile
View File

@@ -1,5 +1,6 @@
# Use NVIDIA CUDA as base image and run the same installation as in the other packages.
FROM nvidia/cuda:10.1-base-ubuntu18.04
# The version of cudatoolkit must match those of the base image, see Dockerfile.pytorch
FROM nvidia/cuda:10.2-base-ubuntu18.04
LABEL maintainer="Jupyter Project <jupyter@googlegroups.com>"
############################################################################
@@ -377,9 +378,9 @@ RUN conda install --quiet --yes \
fix-permissions $CONDA_DIR && \
fix-permissions /home/$NB_USER
# Install PyTorch
# Install PyTorch, version of cudatoolkit must match those of the base image
RUN conda install -y -c pytorch \
cudatoolkit=10.1 \
cudatoolkit=10.2 \
'pytorch=1.3.1' \
torchvision && \
conda clean --all -f -y && \