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