install.sh: fix false-negative GPU-access probe (--entrypoint true)

This commit is contained in:
ent
2026-06-24 14:44:29 +10:00
parent df8981d54c
commit cadfa61184
+2 -2
View File
@@ -158,9 +158,9 @@ verify_host() {
[[ "$FORCE_HW" -eq 1 ]] || die "Not detecting GB10/SM12.1. Pass --force (and maybe --backend) to proceed." [[ "$FORCE_HW" -eq 1 ]] || die "Not detecting GB10/SM12.1. Pass --force (and maybe --backend) to proceed."
warn "Host is not GB10/SM121; proceeding under --force (untested)." warn "Host is not GB10/SM121; proceeding under --force (untested)."
fi fi
# Docker can see the GPU? # Docker can see the GPU? (--entrypoint true so we test GPU access, not the image's CMD)
if ! docker info 2>/dev/null | grep -qiE 'nvidia|Default Runtime: nvidia' \ if ! docker info 2>/dev/null | grep -qiE 'nvidia|Default Runtime: nvidia' \
&& ! docker run --rm --gpus all "$IMAGE" true 2>/dev/null; then && ! docker run --rm --gpus all --entrypoint true "$IMAGE" 2>/dev/null; then
warn "Could not confirm Docker GPU access (nvidia-container-toolkit). 'docker run --gpus all' must work." warn "Could not confirm Docker GPU access (nvidia-container-toolkit). 'docker run --gpus all' must work."
fi fi
# Disk # Disk