Update torch==2.1.0 and torchvision==0.16.0

mrhan1993 · Dec 12, 2023 · 1167e6f · 1167e6f
1 parent ca3c29c
commit 1167e6f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/main.py b/main.py
@@ -226,7 +226,7 @@ def prepare_environments(args) -> bool:
 
         if not is_installed("torch") or not is_installed("torchvision"):
             print(f"torch_index_url: {torch_index_url}")
-            run_pip(f"install torch==2.0.1 torchvision==0.15.2 --extra-index-url {torch_index_url}", "torch")
+            run_pip(f"install torch==2.1.0 torchvision==0.16.0 --extra-index-url {torch_index_url}", "torch")
 
     skip_sync_repo = False
     if args.sync_repo is not None: