specify torch version (sh)

IAHispano · Sep 25, 2024 · 39a38b2 · 39a38b2
1 parent 5a7b5c5
commit 39a38b2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/run-install.sh b/run-install.sh
@@ -86,7 +86,7 @@ create_venv() {
         exit 1
     fi
 
-    python -m pip install torch==2.3.1 torchvision torchaudio --upgrade --index-url https://download.pytorch.org/whl/cu121 
+    python -m pip install torch==2.3.1 torchvision==0.18.1 torchaudio==2.3.1 --upgrade --index-url https://download.pytorch.org/whl/cu121 
     finish
 }