Added the instruction for AMD GPU in Chinese

This commit is contained in:
C3EZ 2025-02-18 13:44:06 +08:00 committed by GitHub
parent b114eefb80
commit e077944d90
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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 设置