Update README.md

This commit is contained in:
Yushen CHEN
2024-12-27 15:21:22 +08:00
committed by GitHub
parent 7370ce7243
commit 6281b4b8ed

View File

@@ -25,19 +25,11 @@
# 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.
# 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
```bash
# Install pytorch with your ROCm version, e.g.
# 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
```