gitea-mirror/F5-TTS
1
0
Fork 0
mirror of https://github.com/SWivid/F5-TTS.git synced 2025-12-31 15:07:55 -08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #671 from Rek-Malorm/main

Browse Source 
Update README.md. Add PyTorch installation instruction for AMD GPU
This commit is contained in:
Yushen CHEN
2024-12-27 15:26:53 +08:00
committed by GitHub
parent 3e73553bd9 6281b4b8ed
commit 20aa6a1103
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
README.md
View File

@@ -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: