diff --git a/images/Dockerfile.nvidia b/images/Dockerfile.nvidia
index d27ca32a1..0b91293df 100644
--- a/images/Dockerfile.nvidia
+++ b/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"]