Skip to content

GitLab

Unverified Commit 659bfd10 authored by Charles Chen's avatar Charles Chen Committed by GitHub
Browse files

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

parent 67e53b16
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
docker/Dockerfile
View file @ 659bfd10
...@@ -8,6 +8,9 @@ ENV DEBIAN_FRONTEND=noninteractive \ ...@@ -8,6 +8,9 @@ ENV DEBIAN_FRONTEND=noninteractive \
CUDA_HOME=/usr/local/cuda \ CUDA_HOME=/usr/local/cuda \
GDRCOPY_HOME=/usr/src/gdrdrv-2.4.4/ \ GDRCOPY_HOME=/usr/src/gdrdrv-2.4.4/ \
NVSHMEM_DIR=/sgl-workspace/nvshmem/install 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 # Set timezone and install all packages
RUN echo 'tzdata tzdata/Areas select America' | debconf-set-selections \ RUN echo 'tzdata tzdata/Areas select America' | debconf-set-selections \
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment