diff --git a/docker-compose.yml b/docker-compose.yml index c6cae1fd..d7317cdb 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -168,14 +168,15 @@ services: - OLLAMA_HOST=0.0.0.0 networks: - MIH-network - # === Added section for NVIDIA GPU acceleration === - # deploy: - # resources: - # reservations: - # devices: - # - driver: nvidia - # count: all # or specify a number of GPUs - # capabilities: [gpu] + # === Added section for NVIDIA GPU acceleration === + runtime: nvidia + deploy: + resources: + reservations: + devices: + - driver: nvidia + count: all # or specify a number of GPUs + capabilities: [ gpu ] #============== Firebaase ==================================================================== # firebase: # container_name: MIH-firebase-emulator