Update README.md

Sanster · Mar 14, 2023 · 9834acd · 9834acd
1 parent de10334
commit 9834acd
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -26,6 +26,8 @@ https://user-images.githubusercontent.com/3998421/223580181-6375fc76-414a-4837-b
 ### Quick Start
 
 ```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 tldream
 tldream --model runwayml/stable-diffusion-v1-5 --device cuda
 ```