chore: increase allowed memory for Triton (#1498)

openfoodfacts · Dec 12, 2024 · 0c52a59 · 0c52a59
1 parent 7fe5324
commit 0c52a59
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/docker/ml-gpu.yml b/docker/ml-gpu.yml
@@ -22,7 +22,7 @@ services:
     # https://github.com/triton-inference-server/server/blob/main/docs/user_guide/model_management.md
     # for more information
     entrypoint: "/opt/nvidia/nvidia_entrypoint.sh tritonserver --model-repository=/models --model-control-mode=explicit --load-model=*"
-    mem_limit: 15g
+    mem_limit: 20g
     runtime: nvidia
     deploy:
       resources:

diff --git a/docker/ml.yml b/docker/ml.yml
@@ -14,7 +14,7 @@ services:
     # https://github.com/triton-inference-server/server/blob/main/docs/user_guide/model_management.md
     # for more information
     entrypoint: "tritonserver --model-repository=/models --model-control-mode=explicit --load-model=*"
-    mem_limit: 15g
+    mem_limit: 20g
 
   fasttext:
     restart: $RESTART_POLICY