From 1176d64b130621617755a0acaf14c991845e53f0 Mon Sep 17 00:00:00 2001 From: oobabooga <112222186+oobabooga@users.noreply.github.com> Date: Sat, 11 Feb 2023 07:56:12 -0300 Subject: [PATCH] Update README.md --- README.md | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/README.md b/README.md index b5adc582..11487998 100644 --- a/README.md +++ b/README.md @@ -42,12 +42,13 @@ pip install -r requirements.txt 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. pip must be used for this because conda instalation is not supported for pytorch yet. - * ``` - pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/rocm5.2 - ``` +* 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, you just need the standard pytorch and should replace the third command with this one: +* If you are running in CPU mode, replace the third command with this one: ``` conda install pytorch torchvision torchaudio git -c pytorch