From 7370ce72434d1fcb12f0c075c3e0fbff0bc341ee Mon Sep 17 00:00:00 2001
From: Rek-Malorm <70321198+Rek-Malorm@users.noreply.github.com>
Date: Fri, 27 Dec 2024 11:26:20 +1300
Subject: [PATCH] Provide instructions for AMD GPU

---
 README.md | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/README.md b/README.md
index c74e016..1d9e162 100644
--- a/README.md
+++ b/README.md
@@ -25,11 +25,22 @@
 # Create a python 3.10 conda env (you could also use virtualenv)
 conda create -n f5-tts python=3.10
 conda activate f5-tts
+```
 
+### Nvidia GPU
+
+```bash
 # 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
+
+```bash
+# 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:
 
 ### 1. As a pip package (if just for inference)