GPT-SoVITS/docker-compose-rocm.yaml
Kaihui-AMD 40a5ea1f66 Add ROCm Docker support for AMD GPUs (RDNA3/RDNA4)
- Add Dockerfile.rocm based on rocm/pytorch:rocm7.2.3
- Add docker-compose-rocm.yaml with /dev/kfd + /dev/dri passthrough
- Add ROCm Docker section to README with quick start and notes
- Use onnxruntime_migraphx from repo.radeon.com for GPU-accelerated ONNX
- Pin starlette<1.0.0 to work around #2762
- Set ROCBLAS_USE_HIPBLASLT=0 for RDNA4 stability

Tested: GPT-SoVITS v2 inference on Radeon AI PRO R9700 (gfx1201)
with ROCm 7.2.3, PyTorch 2.9.1, fp16. WebUI and API both working.
2026-05-29 16:35:53 +08:00

31 lines
575 B
YAML

version: "3.8"
services:
GPT-SoVITS-ROCm:
build:
context: .
dockerfile: Dockerfile.rocm
image: gpt-sovits:rocm7.2.3
container_name: GPT-SoVITS-ROCm
ports:
- "9871:9871"
- "9872:9872"
- "9873:9873"
- "9874:9874"
- "9880:9880"
volumes:
- .:/workspace/GPT-SoVITS
environment:
- is_half=true
- ROCBLAS_USE_HIPBLASLT=0
devices:
- /dev/kfd
- /dev/dri
group_add:
- video
- render
tty: true
stdin_open: true
shm_size: "16g"
restart: unless-stopped