diff --git a/compose.gpu.yml b/compose.gpu.yml index f453134..1989cc7 100644 --- a/compose.gpu.yml +++ b/compose.gpu.yml @@ -18,6 +18,15 @@ services: device_ids: ["0"] capabilities: [gpu] + ollama_research: + deploy: + resources: + reservations: + devices: + - driver: nvidia + device_ids: ["1"] + capabilities: [gpu] + vision: deploy: resources: diff --git a/compose.podman-gpu.yml b/compose.podman-gpu.yml index 688653f..9a947dd 100644 --- a/compose.podman-gpu.yml +++ b/compose.podman-gpu.yml @@ -18,6 +18,14 @@ services: reservations: devices: [] + ollama_research: + devices: + - nvidia.com/gpu=1 + deploy: + resources: + reservations: + devices: [] + vision: devices: - nvidia.com/gpu=0