Updating compose file to include nvidia and rocm containers

ollama/.config/ollama_container/compose.yaml

@@ -1,7 +1,7 @@
 services:
-  llama_server:
+  llama_server_rocm:
     image: ollama/ollama:rocm
-    container_name: llama_server
+    restart: always
     ports:
       - "11434:11434"
     volumes:
@@ -10,3 +10,19 @@ services:
       - /dev/kfd
       - /dev/dri
 
+  llama_server_nvidia:
+    image: ollama/ollama:latest
+    ports:
+      - "11434:11434"
+    volumes:
+      - ./container_data:/root/.ollama:z
+    privileged: true # Had to be done so nvidia can find gpus...
+    restart: always
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: 1 # alternatively, use `count: all` for all GPUs
+              capabilities: [gpu]
+

ollama/.config/ollama_container/compose_nvidia.yaml

@@ -1,14 +0,0 @@
-services:
-  llama_server:
-    image: ollama/ollama:latest
-    container_name: llama_server
-    ports:
-      - "11434:11434"
-    volumes:
-      - ./container_data:/root/.ollama:z
-    privileged: true # Had to be done so nvidia can find gpus...
-    devices:
-      - nvidia.com/gpu=all
-    #   - /dev/kfd
-    #   - /dev/dri
-