diff --git a/README.md b/README.md index 205ba132..b5adc582 100644 --- a/README.md +++ b/README.md @@ -42,7 +42,11 @@ 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. +* 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 are running in CPU mode, you just need the standard pytorch and should replace the third command with this one: ```