From 6281b4b8edd7fd7e8741a4e3282afefc40174fe4 Mon Sep 17 00:00:00 2001 From: Yushen CHEN <45333109+SWivid@users.noreply.github.com> Date: Fri, 27 Dec 2024 15:21:22 +0800 Subject: [PATCH] Update README.md --- README.md | 12 ++---------- 1 file changed, 2 insertions(+), 10 deletions(-) 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 ```