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

Merge remote-tracking branch 'origin/master'

Browse Source
This commit is contained in:
Christoph Schranz
2020-03-11 07:49:11 +01:00
parent 4e0e26e07f db349d928b
commit 46bc7c47ef
2 changed files with 26 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
.build/Dockerfile
View File

@@ -359,28 +359,26 @@ RUN julia -e 'import Pkg; Pkg.update()' && \
LABEL maintainer="Christoph Schranz <christoph.schranz@salzburgresearch.at>"
# Install cudatoolkit must match the version of the base image
# Install Tensorflow, check compatibility here: https://www.tensorflow.org/install/gpu
RUN conda install --quiet --yes \
cudatoolkit=10.1 && \
conda clean --all -f -y && \
fix-permissions $CONDA_DIR && \
fix-permissions /home/$NB_USER
# Install Tensorflow
RUN conda install --quiet --yes \
'tensorflow-gpu' \
'tensorflow-gpu=2.1*' \
'keras-gpu' && \
conda clean --all -f -y && \
fix-permissions $CONDA_DIR && \
fix-permissions /home/$NB_USER
# Install PyTorch with dependencies
RUN conda install --quiet --yes \
pyyaml mkl mkl-include setuptools cmake cffi typing \
pytorch \
torchvision && \
conda clean --all -f -y && \
pip install torch_nightly -f https://download.pytorch.org/whl/nightly/cu90/torch_nightly.html && \
pyyaml mkl mkl-include setuptools cmake cffi typing
# Check compatibility here: https://pytorch.org/get-started/locally/
RUN conda install --quiet --yes \
pytorch \
torchvision \
cudatoolkit=10.1 -c pytorch
# pip install torch_nightly -f https://download.pytorch.org/whl/nightly/cu90/torch_nightly.html && \
# Clean installation
RUN conda clean --all -f -y && \
fix-permissions $CONDA_DIR && \
fix-permissions /home/$NB_USER

28
src/Dockerfile.gpulibs
View File

@@ -1,26 +1,24 @@
LABEL maintainer="Christoph Schranz <christoph.schranz@salzburgresearch.at>"
# Install cudatoolkit must match the version of the base image
# Install Tensorflow, check compatibility here: https://www.tensorflow.org/install/gpu
RUN conda install --quiet --yes \
cudatoolkit=10.1 && \
conda clean --all -f -y && \
fix-permissions $CONDA_DIR && \
fix-permissions /home/$NB_USER
# Install Tensorflow
RUN conda install --quiet --yes \
'tensorflow-gpu' \
'tensorflow-gpu=2.1*' \
'keras-gpu' && \
conda clean --all -f -y && \
fix-permissions $CONDA_DIR && \
fix-permissions /home/$NB_USER
# Install PyTorch with dependencies
RUN conda install --quiet --yes \
pyyaml mkl mkl-include setuptools cmake cffi typing \
pytorch \
torchvision && \
conda clean --all -f -y && \
pip install torch_nightly -f https://download.pytorch.org/whl/nightly/cu90/torch_nightly.html && \
pyyaml mkl mkl-include setuptools cmake cffi typing
# Check compatibility here: https://pytorch.org/get-started/locally/
RUN conda install --quiet --yes \
pytorch \
torchvision \
cudatoolkit=10.1 -c pytorch
# pip install torch_nightly -f https://download.pytorch.org/whl/nightly/cu90/torch_nightly.html && \
# Clean installation
RUN conda clean --all -f -y && \
fix-permissions $CONDA_DIR && \
fix-permissions /home/$NB_USER