Просмотр исходного кода

Merge branch 'main' of github.com:oobabooga/text-generation-webui

oobabooga 3 лет назад
Родитель
Сommit
7eed553337
1 измененных файлов с 7 добавлено и 2 удалено
  1. 7 2
      README.md

+ 7 - 2
README.md

@@ -42,8 +42,13 @@ pip install -r requirements.txt
 
 
 The third line assumes that you have an NVIDIA GPU. 
 The third line assumes that you have an NVIDIA GPU. 
 
 
-* If you have an AMD GPU, you should install the ROCm version of pytorch instead.
-* If you are running in CPU mode, you just need the standard pytorch and should replace the third command with this one:
+* If you have an AMD GPU, replace the third command with this one:
+
+```
+pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/rocm5.2
+```
+  	  
+* If you are running in CPU mode, replace the third command with this one:
 
 
 ```
 ```
 conda install pytorch torchvision torchaudio git -c pytorch
 conda install pytorch torchvision torchaudio git -c pytorch