From b114eefb80532908fee3346d14d4deb3626d7f51 Mon Sep 17 00:00:00 2001 From: C3EZ <96614352+luckykevvv@users.noreply.github.com> Date: Tue, 18 Feb 2025 13:42:17 +0800 Subject: [PATCH] Added the instruction for AMD GPU in English --- README.md | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/README.md b/README.md index a3f5966c..07c1eec7 100644 --- a/README.md +++ b/README.md @@ -115,6 +115,15 @@ brew install ffmpeg pip install -r requirements.txt ``` +##### AMD GPU only + +After installed the dependences above + +```bash +pip3 uninstall torch torchvision torchaudio +pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.2.4 +``` + ### Using Docker #### docker-compose.yaml configuration