From 659bfd10239e284a119bdece95eb502c22dbc943 Mon Sep 17 00:00:00 2001
From: Charles Chen <pychen96@gmail.com>
Date: Wed, 30 Jul 2025 20:28:07 -0700
Subject: [PATCH] Add GKE's default CUDA runtime lib location to PATH and
 LD_LIBRARY_PATH. (#8544)

---
 docker/Dockerfile | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/docker/Dockerfile b/docker/Dockerfile
index 1e3872c9e..7fe76d282 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -8,6 +8,9 @@ ENV DEBIAN_FRONTEND=noninteractive \
     CUDA_HOME=/usr/local/cuda \
     GDRCOPY_HOME=/usr/src/gdrdrv-2.4.4/ \
     NVSHMEM_DIR=/sgl-workspace/nvshmem/install
+# Add GKE default lib and bin locations.
+ENV PATH="${PATH}:/usr/local/nvidia/bin" \
+    LD_LIBRARY_PATH="${LD_LIBRARY_PATH}:/usr/local/nvidia/lib:/usr/local/nvidia/lib64"
 
 # Set timezone and install all packages
 RUN echo 'tzdata tzdata/Areas select America' | debconf-set-selections \