chore: use base image from arg

Signed-off-by: Fred Rolland <[email protected]>

Author: rollandf

images/Dockerfile.nvidia

@@ -12,6 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+ARG BASE_IMAGE_GO_DISTROLESS_DEV
+
 FROM --platform=$BUILDPLATFORM golang:1.23.4 AS builder
 
 ARG GOPROXY
@@ -35,7 +37,7 @@ RUN apt-get update && \
 FROM alpine:3 AS pkgs
 RUN apk add --no-cache hwdata-pci kmod
 
-FROM nvcr.io/nvidia/distroless/go:v3.2.0-dev
+FROM ${BASE_IMAGE_GO_DISTROLESS_DEV:-nvcr.io/nvidia/distroless/go:v3.2.1-dev}
 
 USER 0:0
 SHELL ["/busybox/sh", "-c"]