From c5324d653bfc3eb2043103611628bb27cc243216 Mon Sep 17 00:00:00 2001 From: Spencer-Dawson Date: Sat, 11 Feb 2023 00:13:06 -0700 Subject: [PATCH] re-added missed README changes --- README.md | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) 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: ```