Update README

This commit is contained in:
oobabooga 2023-01-15 16:50:58 -03:00
parent e04ecd4bce
commit 9f1b6e0398

View File

@ -38,7 +38,9 @@ conda activate textgen
conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
``` ```
For AMD GPUs, you need the ROCm version of pytorch. If you don't have any GPU and want to run the webui in CPU mode, you just need the stock pytorch and should use this command instead: For AMD GPUs, you need the ROCm version of pytorch.
If you don't have any GPU and want to run the webui in CPU mode, you just need the stock pytorch and should use this command instead:
``` ```
conda install pytorch torchvision torchaudio -c pytorch conda install pytorch torchvision torchaudio -c pytorch