diff --git a/README.md b/README.md index 1d9e162..3e56efb 100644 --- a/README.md +++ b/README.md @@ -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 ```