Update Dockerfile

Dockerfile

@@ -7,11 +7,11 @@ RUN apt-get update && apt-get install curl -y
     #gnupg
 
 # Add NVIDIA package repositories
-#RUN curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg \
-#    && echo "deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://nvidia.github.io/libnvidia-container/stable/deb/ $(. /etc/os-release; echo $UBUNTU_CODENAME) main" > /etc/apt/sources.list.d/nvidia-container-toolkit.list
+RUN curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg \
+    && echo "deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://nvidia.github.io/libnvidia-container/stable/deb/ $(. /etc/os-release; echo $UBUNTU_CODENAME) main" > /etc/apt/sources.list.d/nvidia-container-toolkit.list
 
 # Install NVIDIA container toolkit (Check for any updated methods or URLs for Ubuntu jammy)
-#RUN apt-get update && apt-get install -y nvidia-container-toolkit || true
+RUN apt-get update && apt-get install -y nvidia-container-toolkit || true
 
 # Install application
 #RUN curl https://ollama.ai/install.sh | sh