build Dockerfile on latest version of docker-stack

2020-02-20 11:03:48 +01:00
parent ac678594e9
commit dd7c6c75ef
8 changed files with 598 additions and 0 deletions
--- a/src/Dockerfile.pytorch
+++ b/src/Dockerfile.pytorch
@@ -0,0 +1,15 @@
+# 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