mirror of
https://github.com/RVC-Boss/GPT-SoVITS.git
synced 2026-06-04 05:01:27 +08:00
- 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.
31 lines
575 B
YAML
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
|