-
Notifications
You must be signed in to change notification settings - Fork 3
Expand file tree
/
Copy pathdocker-compose.rocm.yml
More file actions
28 lines (28 loc) · 1.11 KB
/
Copy pathdocker-compose.rocm.yml
File metadata and controls
28 lines (28 loc) · 1.11 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
# ROCm override:docker compose -f docker-compose.yml -f docker-compose.rocm.yml up -d
# 适用于 AMD GPU(gfx1151/Radeon 8060S 等)+ ROCm 宿主。
# 注意:group_add 的 gid 必须按宿主实际值改(getent group render/video)。
services:
titanvault-minutes:
build:
dockerfile: Dockerfile.rocm
args:
# 宿主已有的 ROCm+PyTorch 镜像(零下载)。各机器不同,按实际改。
ROCM_BASE: cortibox-rocm-base:latest
image: titanvault-minutes:rocm
environment:
- AHAMVOICE_ASR_DEVICE=cuda
# 若报 "no kernel image available" 取消下行注释:
# - HSA_OVERRIDE_GFX_VERSION=11.0.0
# ROCm GPU 透传(/dev/kfd + /dev/dri)。容器进程要在 render 组才能访问。
device_cgroup_rules:
- "c 226:* rmw"
- "c 511:* rmw"
devices:
- /dev/kfd:/dev/kfd
- /dev/dri:/dev/dri
group_add:
# ⚠ 按宿主实际改:getent group render | cut -d: -f3
- "992" # render gid(本机 100.66.1.22)
- "44" # video gid
# ROCm 共享内存需要 host IPC(不设可能 OOM/hang)
ipc: host