Update Dockerfile

Dockerfile

@@ -1,17 +1,19 @@
 # Builder stage
 FROM ubuntu:latest
 
-# Install dependencies
+# Update packages and install curl and gnupg
 RUN apt-get update && apt-get install -y \
     curl \
-    gnupg \
-    && 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 \
-    && apt-get update \
-    && apt-get install -y nvidia-container-toolkit \
-    && rm -rf /var/lib/apt/lists/*
-
-# Install ollama
+    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
+
+# 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
+
+# Install application
 RUN curl https://ollama.ai/install.sh | sh
 
 # Create the directory and give appropriate permissions
@@ -25,8 +27,6 @@ RUN chmod +x /entrypoint.sh
 
 # Set the entry point script as the default command
 ENTRYPOINT ["/entrypoint.sh"]
-
-# Set default arguments for the entry point script
 CMD ["ollama", "serve"]
 
 # Set the model as an environment variable (this can be overridden)