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}"