From 09dc8b454c621b49e0c994b8517ac937d3f52bcc Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Rapha=C3=ABl=20Bournhonesque?=
 <raphael0202@users.noreply.github.com>
Date: Wed, 28 Feb 2024 10:02:51 +0100
Subject: [PATCH] chore: Update docker config (#1321)

* chore: switch to tritonserver 24.01 for GPU

* chore: add the restart_policy to statsd and postgres_exporter services
---
 docker/ml-gpu.yml  | 13 ++-----------
 docker/monitor.yml |  2 ++
 2 files changed, 4 insertions(+), 11 deletions(-)

diff --git a/docker/ml-gpu.yml b/docker/ml-gpu.yml
index a5a60eca53..a6b33d8423 100644
--- a/docker/ml-gpu.yml
+++ b/docker/ml-gpu.yml
@@ -8,17 +8,8 @@ version: "3.9"
 
 services:
   triton:
-    # restart: $RESTART_POLICY
-    # This is a custom built of Triton with:
-    # - GRPC/HTTP support
-    # - GPU support
-    # - Tensorflow and ONNX support
-    # This allows us to reduce the image size
-    # Build with `python3 compose.py --backend tensorflow --backend
-    # onnxruntime`
-    # Release (23.04 version):
-    # https://github.com/triton-inference-server/server/releases/tag/v2.33.0
-    image: ghcr.io/openfoodfacts/triton:23.04-gpu
+    restart: $RESTART_POLICY
+    image: nvcr.io/nvidia/tritonserver:24.01-py3
     ports:
      - ${TRITON_EXPOSE_HTTP:-8000}:8000
      - ${TRITON_EXPOSE_GRPC:-8001}:8001
diff --git a/docker/monitor.yml b/docker/monitor.yml
index c4172eb0d5..b8d8d14a61 100644
--- a/docker/monitor.yml
+++ b/docker/monitor.yml
@@ -11,6 +11,7 @@ services:
     depends_on:
       - statsd
   statsd:
+    restart: $RESTART_POLICY
     image: prom/statsd-exporter:v0.22.2
     volumes:
       - ./statsd.conf:/statsd/statsd.conf
@@ -19,6 +20,7 @@ services:
     ports:
       - ${STATSD_EXPOSE:-9102}:9102
   postgres_exporter:
+    restart: $RESTART_POLICY
     image: prometheuscommunity/postgres-exporter:v0.11.1
     environment:
       - DATA_SOURCE_NAME=postgresql://${POSTGRES_USER}:${POSTGRES_PASSWORD}@${POSTGRES_HOST}:5432/${POSTGRES_DB}?sslmode=disable