Update Dockerfile

Dockerfile

@@ -3,6 +3,18 @@ FROM ubuntu:latest
 
 RUN apt update && apt install curl -y
 
+
+RUN curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey \
+    | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg
+    
+RUN curl -s -L https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list \
+    | sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' \
+    | sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list
+    
+RUN sudo apt-get update
+
+RUN apt-get install -y nvidia-container-toolkit
+
 RUN curl https://ollama.ai/install.sh | sh
 
 # Create the directory and give appropriate permissions