diff --git a/beszel/dockerfile_agent_nvidia b/beszel/dockerfile_agent_nvidia index 77a7a50..e157879 100644 --- a/beszel/dockerfile_agent_nvidia +++ b/beszel/dockerfile_agent_nvidia @@ -15,7 +15,7 @@ RUN CGO_ENABLED=0 GOGC=75 GOOS=$TARGETOS GOARCH=$TARGETARCH go build -ldflags "- # -------------------------- # Final image: GPU-enabled agent with nvidia-smi # -------------------------- -FROM nvidia/cuda:12.9.1-base-ubuntu22.04 +FROM nvidia/cuda:12.2.2-base-ubuntu22.04 COPY --from=builder /agent /agent ENTRYPOINT ["/agent"]