Tentative installation of pytorch

b98943e4 · Nicolas M. Thiéry · cacef72f · b98943e4
Commit b98943e4 authored Oct 18, 2021 by Nicolas M. Thiéry
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

Dockerfile Dockerfile +2 -1

No files found.
--- a/Dockerfile
+++ b/Dockerfile
@@ -83,7 +83,8 @@ RUN jupyter labextension install @jupyterlab/server-proxy
 RUN code-server --install-extension ms-python.python
 RUN pip install git+https://github.com/betatim/vscode-binder.git

-
+# Install pytorch for cpu (conda install fails for now)
+pip install torch==1.9.1+cpu torchvision==0.10.1+cpu torchaudio==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html

 # Install unpackaged jupyterlab extensions
 # run_all_buttons is currently incompatible with latest JupyterLab 3;