Update README.md

ROCm specific installation instruction
ROCm · Oct 3, 2024 · eb986e1 · eb986e1
1 parent d6ea535
commit eb986e1
Showing 1 changed file with 4 additions and 2 deletions.
diff --git 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**: