Merge pull request #671 from Rek-Malorm/main

Update README.md. Add PyTorch installation instruction for AMD GPU
SWivid · Dec 27, 2024 · 20aa6a1 · 20aa6a1
2 parents 3e73553 + 6281b4b
commit 20aa6a1
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -26,8 +26,11 @@
 conda create -n f5-tts python=3.10
 conda activate f5-tts
 
-# Install pytorch with your CUDA version, e.g.
+# NVIDIA GPU: install pytorch with your CUDA version, e.g.
 pip install torch==2.3.0+cu118 torchaudio==2.3.0+cu118 --extra-index-url https://download.pytorch.org/whl/cu118
+
+# AMD GPU: install pytorch with your ROCm version, e.g.
+pip install torch==2.5.1+rocm6.2 torchaudio==2.5.1+rocm6.2 --extra-index-url https://download.pytorch.org/whl/rocm6.2
 ```
 
 Then you can choose from a few options below: