16 lines
512 B
Docker
16 lines
512 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, version of cudatoolkit must match those of the base image
|
|
RUN conda install -y -c pytorch \
|
|
cudatoolkit=10.2 \
|
|
'pytorch=1.3.1' \
|
|
torchvision && \
|
|
conda clean --all -f -y && \
|
|
fix-permissions $CONDA_DIR && \
|
|
fix-permissions /home/$NB_USER
|