2020-02-20 10:03:48 +00:00
|
|
|
# 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
|
|
|
|
|
|
2020-02-22 19:35:24 +00:00
|
|
|
# Install PyTorch, version of cudatoolkit must match those of the base image
|
2020-02-20 10:03:48 +00:00
|
|
|
RUN conda install -y -c pytorch \
|
2020-02-22 19:35:24 +00:00
|
|
|
cudatoolkit=10.2 \
|
2020-02-20 10:03:48 +00:00
|
|
|
'pytorch=1.3.1' \
|
|
|
|
|
torchvision && \
|
|
|
|
|
conda clean --all -f -y && \
|
|
|
|
|
fix-permissions $CONDA_DIR && \
|
|
|
|
|
fix-permissions /home/$NB_USER
|
