From 32d19216e9a07076f12ac110cf02d73f53d1ebe5 Mon Sep 17 00:00:00 2001
From: avouacr <avouacr@pm.me>
Date: Mon, 9 Oct 2023 14:30:29 +0000
Subject: [PATCH] fix(pytorch): fix commands

---
 python-pytorch/Dockerfile | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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} && \