2020-02-26 06:50:27 +00:00
|
|
|
LABEL maintainer="Christoph Schranz <christoph.schranz@salzburgresearch.at>"
|
|
|
|
|
|
2020-03-11 06:48:16 +00:00
|
|
|
# Install Tensorflow, check compatibility here: https://www.tensorflow.org/install/gpu
|
2020-02-24 20:35:15 +00:00
|
|
|
RUN conda install --quiet --yes \
|
2020-03-11 06:48:16 +00:00
|
|
|
'tensorflow-gpu=2.1*' \
|
2020-03-10 17:29:25 +00:00
|
|
|
'keras-gpu' && \
|
2020-02-20 10:03:48 +00:00
|
|
|
fix-permissions $CONDA_DIR && \
|
|
|
|
|
fix-permissions /home/$NB_USER
|
|
|
|
|
|
2020-03-10 17:29:25 +00:00
|
|
|
# Install PyTorch with dependencies
|
|
|
|
|
RUN conda install --quiet --yes \
|
2020-03-11 06:48:16 +00:00
|
|
|
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 && \
|
2020-02-20 10:03:48 +00:00
|
|
|
fix-permissions $CONDA_DIR && \
|
|
|
|
|
fix-permissions /home/$NB_USER
|
