From e880be93d7188932c29b040036731edaf56b0a1a Mon Sep 17 00:00:00 2001
From: ChairC <974833488@qq.com>
Date: Thu, 28 Nov 2024 20:12:52 +0800
Subject: [PATCH] Update: Update requirements.txt.

---
 requirements.txt | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index 53561ff..329a740 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -9,10 +9,12 @@ torch_summary==1.4.5
 tqdm==4.65.0
 pytorch_fid==0.3.0
 flask==3.0.3
+tensorboardX==2.6.1
 
-# If that fails use: pip install torch==1.13.0+cu116 torchvision==0.14.0+cu116 -f https://download.pytorch.org/whl/torch_stable.html
+# If you want to download gpu version
+# Please use: pip install torch==1.13.0+cu116 torchvision==0.14.0+cu116 -f https://download.pytorch.org/whl/torch_stable.html
 # About more torch information please click: https://pytorch.org/get-started/previous-versions/#linux-and-windows-25
 # More versions please click: https://pytorch.org/get-started/previous-versions
 # [Note] torch versions must >= 1.9.0
-torch==1.13.0+cu116
-torchvision==0.14.0+cu116
\ No newline at end of file
+torch>=1.9.0 # More info: https://pytorch.org/get-started/locally/ (recommended)
+torchvision>=0.10.0 # More info: https://pytorch.org/get-started/locally/ (recommended)
\ No newline at end of file