From 1167e6fe3fee70756000c09f07386e2027690467 Mon Sep 17 00:00:00 2001
From: Konie <guofengzuo@hotmail.com>
Date: Tue, 12 Dec 2023 17:54:17 +0800
Subject: [PATCH] Update torch==2.1.0 and torchvision==0.16.0

---
 main.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/main.py b/main.py
index 0329300..388ac26 100644
--- 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: