From 487ff1f7290c2e2d6fa0366521b01964b84accc3 Mon Sep 17 00:00:00 2001
From: kthui <18255193+kthui@users.noreply.github.com>
Date: Tue, 24 Oct 2023 18:44:29 -0700
Subject: [PATCH] Upgrade DCGM from 2.4.7 to 3.2.6

---
 Dockerfile.sdk | 2 +-
 build.py       | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/Dockerfile.sdk b/Dockerfile.sdk
index 1881f6c0e0..33cebf07a8 100644
--- a/Dockerfile.sdk
+++ b/Dockerfile.sdk
@@ -42,7 +42,7 @@ ARG JAVA_BINDINGS_MAVEN_VERSION=3.8.4
 ARG JAVA_BINDINGS_JAVACPP_PRESETS_TAG=1.5.8
 
 # DCGM version to install for Model Analyzer
-ARG DCGM_VERSION=2.4.7
+ARG DCGM_VERSION=3.2.6
 
 ARG NVIDIA_TRITON_SERVER_SDK_VERSION=unknown
 ARG NVIDIA_BUILD_ID=unknown
diff --git a/build.py b/build.py
index 1c98c2f00a..ca47779e74 100755
--- a/build.py
+++ b/build.py
@@ -76,7 +76,7 @@
         "1.16.0",  # ORT
         "2023.0.0",  # ORT OpenVINO
         "2023.0.0",  # Standalone OpenVINO
-        "2.4.7",  # DCGM version
+        "3.2.6",  # DCGM version
         "py310_23.1.0-1",  # Conda version
         "0.2.1.post1",  # vLLM version
     )