Add GKE's default CUDA runtime lib location to PATH and LD_LIBRARY_PATH. (#8544)

659bfd10 · Charles Chen · GitHub · 67e53b16 · 659bfd10
Unverified Commit 659bfd10 authored Jul 30, 2025 by Charles Chen Committed by GitHub Jul 30, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

docker/Dockerfile docker/Dockerfile +3 -0

No files found.
--- 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 \