install.sh: fix false-negative GPU-access probe (--entrypoint true)
This commit is contained in:
+2
-2
@@ -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
|
||||||
|
|||||||
Reference in New Issue
Block a user