Update redundant headers and declaration clobbering

This commit is contained in:
2026-01-22 11:57:19 -05:00
parent 32e40d3a7f
commit 3c14c8188a

View File

@@ -8,7 +8,6 @@ IMAGE_TAG="${TENTACLE_TAG:-latest}"
# -----------------------------
REGISTRY="hub.krkn.tech"
IMAGE_BASE="hub.krkn.tech/krkncli/cuda"
IMAGE_TAG="latest"
CONTAINER_NAME="tentacle"
SUPPORTED_CUDA=("12.4.1" "12.5.1" "12.6.3" "13.0.1" "13.1.0")
@@ -88,35 +87,22 @@ fi
DETECTED_CUDA="$(nvidia-smi | grep 'CUDA Version' | sed 's/.*CUDA Version: *\([0-9.]*\).*/\1/')"
log "Detected CUDA capability: ${DETECTED_CUDA}"
choose_cuda() {
choose_cuda() {
# Pick highest supported version <= detected CUDA capability
for v in $(printf '%s\n' "${SUPPORTED_CUDA[@]}" | sort -rV); do
if [ "$(printf '%s\n%s\n' "$v" "$DETECTED_CUDA" | sort -V | head -n1)" = "$v" ]; then
echo "$v"
return
return 0
fi
done
echo ""
}
# fallback: highest <= detected
if [ -z "$CUDA_VERSION" ]; then
warn "Could not auto-match CUDA version, defaulting to highest supported"
CUDA_VERSION="$(printf '%s\n' "${SUPPORTED_CUDA[@]}" | sort -V | tail -n1)"
else
log "Selected CUDA image version: ${CUDA_VERSION}"
fi
echo ""
return 1
}
CUDA_VERSION="$(choose_cuda)"
if [ -z "$CUDA_VERSION" ]; then
warn "Could not auto-match CUDA version, defaulting to latest (12.6.3)"
CUDA_VERSION="12.6.3"
else
if CUDA_VERSION="$(choose_cuda)"; then
log "Selected CUDA image version: ${CUDA_VERSION}"
else
warn "Could not auto-match CUDA version, defaulting to highest supported"
CUDA_VERSION="$(printf '%s\n' "${SUPPORTED_CUDA[@]}" | sort -V | tail -n1)"
fi
IMAGE="${IMAGE_BASE}/${CUDA_VERSION}/tentacle:${IMAGE_TAG}"