Files
gpu-jupyter/src/Dockerfile.pytorch
2020-02-20 11:03:48 +01:00

16 lines
453 B
Docker

# Install PyTorch dependencies
RUN conda install --quiet --yes \
pyyaml mkl mkl-include setuptools cmake cffi typing && \
conda clean --all -f -y && \
fix-permissions $CONDA_DIR && \
fix-permissions /home/$NB_USER
# Install PyTorch
RUN conda install -y -c pytorch \
cudatoolkit=10.1 \
'pytorch=1.3.1' \
torchvision && \
conda clean --all -f -y && \
fix-permissions $CONDA_DIR && \
fix-permissions /home/$NB_USER