Merge pull request #82 from shifa-khan/69

Add CUDA_HOME env var to CUDA base images

Author: LalatenduMohanty

Containerfile.cuda.template

@@ -65,6 +65,7 @@ WORKDIR /opt/app-root/bin
 # -----------------------------------------------------------------------------
 ENV NVIDIA_REQUIRE_CUDA=${NVIDIA_REQUIRE_CUDA}
 ENV CUDA_VERSION=${CUDA_VERSION}
+ENV CUDA_HOME=/usr/local/cuda
 ENV NV_CUDA_CUDART_VERSION=${NV_CUDA_CUDART_VERSION}
 ENV NV_CUDA_LIB_VERSION=${NV_CUDA_LIB_VERSION}
 ENV NV_NVTX_VERSION=${NV_NVTX_VERSION}

cuda/12.8/Containerfile

@@ -65,6 +65,7 @@ WORKDIR /opt/app-root/bin
 # -----------------------------------------------------------------------------
 ENV NVIDIA_REQUIRE_CUDA=${NVIDIA_REQUIRE_CUDA}
 ENV CUDA_VERSION=${CUDA_VERSION}
+ENV CUDA_HOME=/usr/local/cuda
 ENV NV_CUDA_CUDART_VERSION=${NV_CUDA_CUDART_VERSION}
 ENV NV_CUDA_LIB_VERSION=${NV_CUDA_LIB_VERSION}
 ENV NV_NVTX_VERSION=${NV_NVTX_VERSION}

cuda/12.9/Containerfile

@@ -65,6 +65,7 @@ WORKDIR /opt/app-root/bin
 # -----------------------------------------------------------------------------
 ENV NVIDIA_REQUIRE_CUDA=${NVIDIA_REQUIRE_CUDA}
 ENV CUDA_VERSION=${CUDA_VERSION}
+ENV CUDA_HOME=/usr/local/cuda
 ENV NV_CUDA_CUDART_VERSION=${NV_CUDA_CUDART_VERSION}
 ENV NV_CUDA_LIB_VERSION=${NV_CUDA_LIB_VERSION}
 ENV NV_NVTX_VERSION=${NV_NVTX_VERSION}

cuda/13.0/Containerfile

@@ -65,6 +65,7 @@ WORKDIR /opt/app-root/bin
 # -----------------------------------------------------------------------------
 ENV NVIDIA_REQUIRE_CUDA=${NVIDIA_REQUIRE_CUDA}
 ENV CUDA_VERSION=${CUDA_VERSION}
+ENV CUDA_HOME=/usr/local/cuda
 ENV NV_CUDA_CUDART_VERSION=${NV_CUDA_CUDART_VERSION}
 ENV NV_CUDA_LIB_VERSION=${NV_CUDA_LIB_VERSION}
 ENV NV_NVTX_VERSION=${NV_NVTX_VERSION}

cuda/13.1/Containerfile

@@ -65,6 +65,7 @@ WORKDIR /opt/app-root/bin
 # -----------------------------------------------------------------------------
 ENV NVIDIA_REQUIRE_CUDA=${NVIDIA_REQUIRE_CUDA}
 ENV CUDA_VERSION=${CUDA_VERSION}
+ENV CUDA_HOME=/usr/local/cuda
 ENV NV_CUDA_CUDART_VERSION=${NV_CUDA_CUDART_VERSION}
 ENV NV_CUDA_LIB_VERSION=${NV_CUDA_LIB_VERSION}
 ENV NV_NVTX_VERSION=${NV_NVTX_VERSION}