update nvidia driver installation to match mwt2 puppet

Dockerfile

@@ -33,15 +33,17 @@ RUN yum install -y \
   man-db \
   telnet 
 
-RUN yum install http://mirror.grid.uchicago.edu/pub/mwt2/sw/el7/HEP_OSlibs-7.2.9-1.el7.cern.x86_64.rpm -y
-
 # Install GPU libraries
-RUN yum install -y https://developer.download.nvidia.com/compute/cuda/repos/rhel7/x86_64/cuda-repo-rhel7-10.2.89-1.x86_64.rpm
-RUN rpm --import http://developer.download.nvidia.com/compute/cuda/repos/rhel7/x86_64/7fa2af80.pub
-RUN rpm --import http://developer.download.nvidia.com/compute/cuda/repos/rhel7/x86_64/D42D0685.pub
-# These match the NVIDIA settings on the AF as of Oct 2023
-RUN yum localinstall -y https://developer.download.nvidia.com/compute/cuda/repos/rhel7/x86_64/nvidia-driver-branch-535-535.86.10-1.el7.x86_64.rpm
-RUN yum install -y cuda-12-2
+COPY repo/cuda.repo /etc/yum.repos.d/cuda.repo
+COPY repo/nvidia.repo /etc/yum.repos.d/nvidia.repo
+RUN yum install cuda-12-2 nvidia-driver-latest-dkms -y
+
+#RUN yum install -y https://developer.download.nvidia.com/compute/cuda/repos/rhel7/x86_64/cuda-repo-rhel7-10.2.89-1.x86_64.rpm
+#RUN rpm --import http://developer.download.nvidia.com/compute/cuda/repos/rhel7/x86_64/7fa2af80.pub
+#RUN rpm --import http://developer.download.nvidia.com/compute/cuda/repos/rhel7/x86_64/D42D0685.pub
+## These match the NVIDIA settings on the AF as of Oct 2023
+#RUN yum localinstall -y https://developer.download.nvidia.com/compute/cuda/repos/rhel7/x86_64/nvidia-driver-branch-535-535.86.10-1.el7.x86_64.rpm
+#RUN yum install -y cuda-12-2
 
 RUN yum install --enablerepo=osg-upcoming -y condor
 

repo/cuda.repo

@@ -0,0 +1,6 @@
+[cuda-rhel7-x86_64]
+name=cuda-rhel7-x86_64
+baseurl=https://developer.download.nvidia.com/compute/cuda/repos/rhel7/x86_64
+enabled=1
+gpgcheck=1
+gpgkey=https://developer.download.nvidia.com/compute/cuda/repos/rhel7/x86_64/D42D0685.pub

repo/nvidia.repo

@@ -0,0 +1,29 @@
+[libnvidia-container]
+name=libnvidia-container
+baseurl=https://nvidia.github.io/libnvidia-container/stable/centos7/$basearch
+repo_gpgcheck=1
+gpgcheck=0
+enabled=1
+gpgkey=https://nvidia.github.io/libnvidia-container/gpgkey
+sslverify=1
+sslcacert=/etc/pki/tls/certs/ca-bundle.crt
+
+[nvidia-container-runtime]
+name=nvidia-container-runtime
+baseurl=https://nvidia.github.io/nvidia-container-runtime/centos7/$basearch
+repo_gpgcheck=1
+gpgcheck=0
+enabled=1
+gpgkey=https://nvidia.github.io/nvidia-container-runtime/gpgkey
+sslverify=1
+sslcacert=/etc/pki/tls/certs/ca-bundle.crt
+
+[nvidia-docker]
+name=nvidia-docker
+baseurl=https://nvidia.github.io/nvidia-docker/centos7/$basearch
+repo_gpgcheck=1
+gpgcheck=0
+enabled=1
+gpgkey=https://nvidia.github.io/nvidia-docker/gpgkey
+sslverify=1
+sslcacert=/etc/pki/tls/certs/ca-bundle.crt