diff --git a/readme.md b/readme.md index 9f6491f9..76e2af03 100644 --- a/readme.md +++ b/readme.md @@ -162,7 +162,12 @@ Or if you want to open a remote port, use ### Linux (AMD GPUs) -Same with the above instructions. Not intensively tested, however. +Same with the above instructions. You need to change torch to AMD version + + pip uninstall torch torchvision torchaudio torchtext functorch xformers + pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm5.6 + +AMD is not intensively tested, however. ### Mac/Windows(AMD GPUs)