fix(pytorch): cpu install command has changed

InseeFrLab · Oct 9, 2023 · 9412561 · 9412561
1 parent 162b108
commit 9412561
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python-pytorch/Dockerfile b/python-pytorch/Dockerfile
@@ -7,10 +7,10 @@ USER root
 
 RUN if ! [[ -z "${CUDA_VERSION}" ]]; then \
         # CUDA support
-        pip install torch torchvision torchaudio; \
+        pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118; \
     else \
         # CPU only
-        pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118; \
+        pip3 install torch torchvision torchaudio; \
     fi && \
     # Fix permissions
     chown -R ${USERNAME}:${GROUPNAME} ${HOME} ${MAMBA_DIR} && \