From 907fadfb6de037d211f6dabd0d73fab844f9ddc2 Mon Sep 17 00:00:00 2001
From: Stephen Nicholas Swatman <stephen.nicholas.swatman@cern.ch>
Date: Tue, 29 Oct 2024 16:41:10 +0100
Subject: [PATCH] Fix Ubuntu 24.04 CUDA environment variables

I forgot to set these correctly after the last update.
---
 ubuntu2404_cuda/Dockerfile | 19 +++++++++----------
 1 file changed, 9 insertions(+), 10 deletions(-)

diff --git a/ubuntu2404_cuda/Dockerfile b/ubuntu2404_cuda/Dockerfile
index 583c8ba..0d1ad36 100644
--- a/ubuntu2404_cuda/Dockerfile
+++ b/ubuntu2404_cuda/Dockerfile
@@ -23,13 +23,12 @@ RUN apt-get update && \
                        cuda-minimal-build-${CUDA_VERSION} &&                   \
     apt-get clean -y
 
-ENV CUDA_PATH "${CUDA_INSTALL_PATH}"
-ENV CUDA_ROOT "${CUDA_INSTALL_PATH}"
-ENV CUDA_HOME "${CUDA_INSTALL_PATH}"
-ENV CUDACXX "${CUDA_INSTALL_PATH}/bin/nvcc"
-ENV CUDACC "${CUDA_INSTALL_PATH}/bin/nvcc"
-
-ENV PATH "${CUDA_INSTALL_PATH}/bin:${PATH}"
-ENV MANPATH "${CUDA_INSTALL_PATH}/share/man:${MANPATH}"
-ENV INCLUDE "${CUDA_INSTALL_PATH}/include:${INCLUDE}"
-ENV LD_LIBRARY_PATH "${CUDA_INSTALL_PATH}/lib64:${CUDA_INSTALL_PATH}/lib:${LD_LIBRARY_PATH}"
+ENV CUDA_PATH "/usr/local/cuda/"
+ENV CUDA_ROOT "${CUDA_PATH}"
+ENV CUDA_HOME "${CUDA_PATH}"
+ENV CUDACXX "${CUDA_PATH}/bin/nvcc"
+ENV CUDACC "${CUDA_PATH}/bin/nvcc"
+ENV NVIDIA_VISIBLE_DEVICES all
+ENV NVIDIA_DRIVER_CAPABILITIES compute,utility
+ENV PATH "${CUDA_PATH}/bin:${PATH}"
+ENV LD_LIBRARY_PATH "${CUDA_PATH}/lib64:${LD_LIBRARY_PATH}"