Update install.py

dill-shower · Nov 7, 2024 · 9113014 · 9113014
1 parent e4e444f
commit 9113014
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/install.py b/install.py
@@ -44,7 +44,7 @@ def prepare_environment():
         "torch",
         (
             f"{sys.executable} -m pip install torch torchvision torchaudio "
-            "--index-url https://download.pytorch.org/whl/cu118"
+            "--index-url https://download.pytorch.org/whl/cu121"
         ),
     )