From 19141ff5c99d0d3f24c63350765e3d1c34f9942b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=A4=A9=E6=B6=AF=E5=8F=A4=E5=B7=B7?= <flavioy@live.com>
Date: Tue, 2 Jan 2024 14:37:31 +0800
Subject: [PATCH] Update README.md

---
 README.md | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/README.md b/README.md
index d7cf581..86b6b85 100755
--- a/README.md
+++ b/README.md
@@ -137,12 +137,12 @@ conda activate videoEnv
 
 - 安装GPU版本Pytorch:
       
-  ```shell 
-  conda install pytorch==2.0.1 torchvision==0.15.2 pytorch-cuda=11.7 -c pytorch -c nvidia
+  ```shell
+  conda install pytorch==2.1.0 torchvision==0.16.0 pytorch-cuda=11.8 -c pytorch -c nvidia
   ```
   或者使用
-  ```shell 
-  pip install torch==2.0.1 torchvision==0.15.2 --index-url https://download.pytorch.org/whl/cu117
+  ```shell
+  pip install torch==2.1.0 torchvision==0.16.0 --index-url https://download.pytorch.org/whl/cu118
   ```
 
 - 安装其他依赖:
@@ -214,6 +214,10 @@ LAMA_SUPER_FAST = False  # 保证效果
 
 解决方案：改用cuda 11.8
 
+```shell
+pip install torch==2.1.0 torchvision==0.16.0 --index-url https://download.pytorch.org/whl/cu118
+```
+
 ## 赞助
 <img src="https://i.imgur.com/EMCP5Lv.jpeg" width="600">