From d1f36dde8eb35cbb789e751ae540c260ec7f2006 Mon Sep 17 00:00:00 2001
From: Qing <cwq1913@gmail.com>
Date: Tue, 28 Feb 2023 21:43:11 +0800
Subject: [PATCH] Update README.md

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

diff --git a/README.md b/README.md
index aee523e..7324bc7 100644
--- a/README.md
+++ b/README.md
@@ -57,6 +57,8 @@ https://user-images.githubusercontent.com/3998421/196976498-ba1ad3ab-fa18-4c55-9
 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 lama-cleaner
 lama-cleaner --model=lama --device=cpu --port=8080
 ```