diff --git a/Dockerfile.sdk b/Dockerfile.sdk index 7538084c31..46ed8c8deb 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 ed9ed535fd..901ca80f5f 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 )