From d283595681e8293d92e6163a4a13d54f5b902e10 Mon Sep 17 00:00:00 2001
From: Qing <cwq1913@gmail.com>
Date: Mon, 10 Apr 2023 09:16:35 +0800
Subject: [PATCH] Update README.md

---
 README.md | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 83c0dd6..7c12a78 100644
--- a/README.md
+++ b/README.md
@@ -61,7 +61,7 @@ Lama Cleaner make it easy to use SOTA AI model in just two commands:
 
 ```bash
 # In order to use the GPU, install cuda version of pytorch first.
-# pip install torch==1.13.1+cu117 --extra-index-url https://download.pytorch.org/whl/cu117
+# pip install torch==1.13.1+cu117 torchvision==0.14.1 --extra-index-url https://download.pytorch.org/whl/cu117
 pip install lama-cleaner
 lama-cleaner --model=lama --device=cpu --port=8080
 ```