From eb986e142133c4e806760a85e90d3e693fe4e7e1 Mon Sep 17 00:00:00 2001 From: Max Podkorytov <4273004+tenpercent@users.noreply.github.com> Date: Wed, 2 Oct 2024 21:18:51 -0700 Subject: [PATCH] Update README.md ROCm specific installation instruction --- README.md | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index f53bdb5607..7867b8ef4d 100644 --- a/README.md +++ b/README.md @@ -38,9 +38,11 @@ conda install xformers -c xformers ```bash # cuda 11.8 version -pip3 install -U xformers --index-url https://download.pytorch.org/whl/cu118 +python -m pip install -U xformers --index-url https://download.pytorch.org/whl/cu118 # cuda 12.1 version -pip3 install -U xformers --index-url https://download.pytorch.org/whl/cu121 +python -m pip install -U xformers --index-url https://download.pytorch.org/whl/cu121 +# rocm 6.1 version (linux only) +python -m pip install -U xformers --index-url https://download.pytorch.org/whl/rocm6.1 ``` * **Development binaries**: