From 39a38b243243f45b4ab2aa8b629d5bac960364de Mon Sep 17 00:00:00 2001 From: Pascal Aznar Date: Wed, 25 Sep 2024 22:47:18 +0200 Subject: [PATCH] specify torch version (sh) --- run-install.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/run-install.sh b/run-install.sh index 3ac48b93..96a1514c 100644 --- 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 }