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
 ```