From 3e7e915c1615ab692c2eb7789f2fe6229db93fc1 Mon Sep 17 00:00:00 2001
From: gujing <zibai.gj@alibaba-inc.com>
Date: Sun, 4 Feb 2024 14:55:16 +0800
Subject: [PATCH] update tritonserver base image to
 nvcr.io/nvidia/tritonserver:24.01-py3 (#1036)

---
 charts/triton/values.yaml         | 2 +-
 docs/serving/triton/serving.md    | 2 +-
 pkg/argsbuilder/serving_triton.go | 2 +-
 3 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/charts/triton/values.yaml b/charts/triton/values.yaml
index 2395245bf..d0745f932 100755
--- a/charts/triton/values.yaml
+++ b/charts/triton/values.yaml
@@ -11,7 +11,7 @@ serviceType: ClusterIP
 servingName:
 servingVersion:
 
-image: "nvcr.io/nvidia/tritonserver:20.12-py3"
+image: "nvcr.io/nvidia/tritonserver:24.01-py3"
 
 imagePullPolicy: "IfNotPresent"
 
diff --git a/docs/serving/triton/serving.md b/docs/serving/triton/serving.md
index 1f7ad675e..df87d7290 100644
--- a/docs/serving/triton/serving.md
+++ b/docs/serving/triton/serving.md
@@ -9,7 +9,7 @@ $ arena serve triton \
  --name=test-triton \
  --namespace=triton \
  --gpus=1 \
- --image=nvcr.io/nvidia/tritonserver:20.12-py3 \
+ --image=nvcr.io/nvidia/tritonserver:24.01-py3 \
  --data=triton-pvc:/mnt/models \
  --model-repository=/mnt/models/ai/triton/model_repository
  
diff --git a/pkg/argsbuilder/serving_triton.go b/pkg/argsbuilder/serving_triton.go
index 69983c914..d8b388709 100644
--- a/pkg/argsbuilder/serving_triton.go
+++ b/pkg/argsbuilder/serving_triton.go
@@ -24,7 +24,7 @@ import (
 )
 
 const (
-	DefaultTritonServingImage = "nvcr.io/nvidia/tritonserver:20.12-py3"
+	DefaultTritonServingImage = "nvcr.io/nvidia/tritonserver:24.01-py3"
 )
 
 type TritonServingArgsBuilder struct {