diff --git a/python-pytorch/Dockerfile b/python-pytorch/Dockerfile
index c340e15e..6c5f729e 100644
--- a/python-pytorch/Dockerfile
+++ b/python-pytorch/Dockerfile
@@ -7,10 +7,10 @@ USER root
 
 RUN if ! [[ -z "${CUDA_VERSION}" ]]; then \
         # CUDA support
-        pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118; \
+        pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118; \
     else \
         # CPU only
-        pip3 install torch torchvision torchaudio; \
+        pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu; \
     fi && \
     # Fix permissions
     chown -R ${USERNAME}:${GROUPNAME} ${HOME} ${MAMBA_DIR} && \