ollama tweaks

ollama/compose-gpu.yaml

@@ -1,15 +1,19 @@
 # Ollama GPU
 services:
   ollama:
-    image: ollama/ollama
+    image: ${OLLAMA_IMAGE:-ollama/ollama}
     volumes:
       - ${VOL_PATH}/ollama:/root/.ollama
-    runtime: nvidia
-    environment:
-      - NVIDIA_VISIBLE_DEVICES=all
-      - NVIDIA_DRIVER_CAPABILITIES=compute,utility
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: all
+              capabilities: [gpu]
     ports:
       - "11434:11434"
+    restart: unless-stopped
 
   webui:
     image: ${WEBUI_IMAGE:-ghcr.io/open-webui/open-webui:main}