From e077944d90e10c38d65b602aa09ab010da5f7ac2 Mon Sep 17 00:00:00 2001 From: C3EZ <96614352+luckykevvv@users.noreply.github.com> Date: Tue, 18 Feb 2025 13:44:06 +0800 Subject: [PATCH] Added the instruction for AMD GPU in Chinese --- docs/cn/README.md | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/docs/cn/README.md b/docs/cn/README.md index a063128..82f5d05 100644 --- a/docs/cn/README.md +++ b/docs/cn/README.md @@ -112,6 +112,15 @@ brew install ffmpeg pip install -r requirements.txt ``` +##### 使用AMD显卡 + +在安装完依赖以后 + +```bash +pip3 uninstall torch torchvision torchaudio +pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.2.4 +``` + ### 在 Docker 中使用 #### docker-compose.yaml 设置