mirror of
https://github.com/jamiepine/voicebox.git
synced 2026-09-12 19:30:38 -07:00
Mike KeyandGitHub
c2282b256a
* Fix ROCm setup for Linux AMD GPUs - Ensure Docker ROCm builds resolve PyTorch packages from the ROCm wheel index so later dependency installs do not replace them with CUDA wheels. - Move ROCm device group handling to a runtime entrypoint that joins the groups owning /dev/kfd and /dev/dri, avoiding distro-specific render/video GID defaults. - Leave HSA_OVERRIDE_GFX_VERSION unset by default in the ROCm compose overlay so newer RDNA GPUs can use native ROCm detection. - Add Linux GPU detection to the Unix setup recipe so AMD systems install ROCm torch wheels and NVIDIA systems install CUDA wheels before backend dependencies. * docs(changelog): add Linux ROCm setup entry * fix(setup): pin ROCm torch wheels and prefer NVIDIA over amdgpu - Install torch/torchaudio from the ROCm index only, before the pooled requirements install, so a plain PyPI (CUDA) wheel can't outrank +rocm - Detect NVIDIA before AMD and gate ROCm on /dev/kfd, so hybrid AMD+NVIDIA hosts get CUDA instead of ROCm
16 lines
462 B
Bash
16 lines
462 B
Bash
#!/bin/sh
|
|
set -e
|
|
# Join whatever groups own the mounted GPU nodes so /dev/kfd and /dev/dri work
|
|
# on any host (no RENDER_GID/VIDEO_GID needed), then drop to the app user.
|
|
for dev in /dev/kfd /dev/dri/render*; do
|
|
[ -e "$dev" ] || continue
|
|
gid=$(stat -c %g "$dev")
|
|
grp=$(getent group "$gid" | cut -d: -f1)
|
|
[ -n "$grp" ] || {
|
|
grp="gpu$gid"
|
|
groupadd -g "$gid" "$grp"
|
|
}
|
|
usermod -aG "$grp" voicebox
|
|
done
|
|
exec gosu voicebox "$@"
|