diff --git a/Dockerfile.nvidia b/Dockerfile.nvidia
index e1ea9e3e..97cfaaeb 100644
--- a/Dockerfile.nvidia
+++ b/Dockerfile.nvidia
@@ -1,3 +1,5 @@
+ARG BASE_IMAGE_GO_DISTROLESS_DEV
+
 FROM golang:1.24-alpine as builder
 
 ARG GOPROXY
@@ -13,7 +15,7 @@ RUN apk add --no-cache --virtual build-dependencies build-base=~0.5 && \
     make clean && \
     make build
 
-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"]