From 85897d742aac4e5d03808bbd7742f9e337c6f543 Mon Sep 17 00:00:00 2001 From: Yingge He Date: Fri, 25 Jul 2025 18:31:01 -0700 Subject: [PATCH 1/2] Anchor datacenter-gpu-manager-4-core version --- Dockerfile.sdk | 4 +++- build.py | 12 ++++++++---- 2 files changed, 11 insertions(+), 5 deletions(-) diff --git a/Dockerfile.sdk b/Dockerfile.sdk index 7d114e2ce5..94d8fd8a46 100644 --- a/Dockerfile.sdk +++ b/Dockerfile.sdk @@ -265,7 +265,9 @@ RUN if [ "$TRITON_ENABLE_GPU" = "ON" ]; then \ curl -o /tmp/cuda-keyring.deb \ https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/$arch/cuda-keyring_1.1-1_all.deb \ && apt install /tmp/cuda-keyring.deb && rm /tmp/cuda-keyring.deb && \ - apt update && apt install --yes datacenter-gpu-manager-4-dev=1:${DCGM_VERSION}; \ + apt update && \ + apt install --yes datacenter-gpu-manager-4-core=1:${DCGM_VERSION} && \ + apt install --yes datacenter-gpu-manager-4-dev=1:${DCGM_VERSION}; \ fi # Build expects "python" executable (not python3). diff --git a/build.py b/build.py index 230b4dc88c..f8131d7075 100755 --- a/build.py +++ b/build.py @@ -858,9 +858,10 @@ def install_dcgm_libraries(dcgm_version, target_machine): # Install DCGM. Steps from https://developer.nvidia.com/dcgm#Downloads RUN dnf config-manager --add-repo https://developer.download.nvidia.com/compute/cuda/repos/rhel8/sbsa/cuda-rhel8.repo \\ && dnf clean expire-cache \\ + && dnf install --assumeyes datacenter-gpu-manager-4-core=1:{} \\ && dnf install --assumeyes datacenter-gpu-manager-4-devel=1:{} """.format( - dcgm_version, dcgm_version + dcgm_version, dcgm_version, dcgm_version ) else: return """ @@ -868,9 +869,10 @@ def install_dcgm_libraries(dcgm_version, target_machine): # Install DCGM. Steps from https://developer.nvidia.com/dcgm#Downloads RUN dnf config-manager --add-repo https://developer.download.nvidia.com/compute/cuda/repos/rhel8/x86_64/cuda-rhel8.repo \\ && dnf clean expire-cache \\ + && dnf install --assumeyes datacenter-gpu-manager-4-core=1:{} \\ && dnf install --assumeyes datacenter-gpu-manager-4-devel=1:{} """.format( - dcgm_version, dcgm_version + dcgm_version, dcgm_version, dcgm_version ) else: if target_machine == "aarch64": @@ -882,9 +884,10 @@ def install_dcgm_libraries(dcgm_version, target_machine): && apt install /tmp/cuda-keyring.deb \\ && rm /tmp/cuda-keyring.deb \\ && apt update \\ + && apt install --yes datacenter-gpu-manager-4-core=1:{} \\ && apt install --yes datacenter-gpu-manager-4-dev=1:{} """.format( - dcgm_version, dcgm_version + dcgm_version, dcgm_version, dcgm_version ) else: return """ @@ -895,9 +898,10 @@ def install_dcgm_libraries(dcgm_version, target_machine): && apt install /tmp/cuda-keyring.deb \\ && rm /tmp/cuda-keyring.deb \\ && apt update \\ + && apt install --yes datacenter-gpu-manager-4-core=1:{} \\ && apt install --yes datacenter-gpu-manager-4-dev=1:{} """.format( - dcgm_version, dcgm_version + dcgm_version, dcgm_version, dcgm_version ) From 061fbf51757a16f648907145e0ca3f84866fe9f8 Mon Sep 17 00:00:00 2001 From: Yingge He <157551214+yinggeh@users.noreply.github.com> Date: Mon, 28 Jul 2025 14:59:27 -0700 Subject: [PATCH 2/2] Apply suggestions from code review Co-authored-by: Misha Chornyi <99709299+mc-nv@users.noreply.github.com> --- Dockerfile.sdk | 5 +++-- build.py | 20 ++++++++++++-------- 2 files changed, 15 insertions(+), 10 deletions(-) diff --git a/Dockerfile.sdk b/Dockerfile.sdk index 94d8fd8a46..8fbc8cd3c0 100644 --- a/Dockerfile.sdk +++ b/Dockerfile.sdk @@ -266,8 +266,9 @@ RUN if [ "$TRITON_ENABLE_GPU" = "ON" ]; then \ https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/$arch/cuda-keyring_1.1-1_all.deb \ && apt install /tmp/cuda-keyring.deb && rm /tmp/cuda-keyring.deb && \ apt update && \ - apt install --yes datacenter-gpu-manager-4-core=1:${DCGM_VERSION} && \ - apt install --yes datacenter-gpu-manager-4-dev=1:${DCGM_VERSION}; \ + apt install --yes --no-install-recommends \ + datacenter-gpu-manager-4-core=1:${DCGM_VERSION} \ + datacenter-gpu-manager-4-dev=1:${DCGM_VERSION}; \ fi # Build expects "python" executable (not python3). diff --git a/build.py b/build.py index f8131d7075..6c726e6c87 100755 --- a/build.py +++ b/build.py @@ -858,8 +858,9 @@ def install_dcgm_libraries(dcgm_version, target_machine): # Install DCGM. Steps from https://developer.nvidia.com/dcgm#Downloads RUN dnf config-manager --add-repo https://developer.download.nvidia.com/compute/cuda/repos/rhel8/sbsa/cuda-rhel8.repo \\ && dnf clean expire-cache \\ - && dnf install --assumeyes datacenter-gpu-manager-4-core=1:{} \\ - && dnf install --assumeyes datacenter-gpu-manager-4-devel=1:{} + && dnf install --assumeyes \\ + datacenter-gpu-manager-4-core=1:{} \\ + datacenter-gpu-manager-4-devel=1:{} """.format( dcgm_version, dcgm_version, dcgm_version ) @@ -869,8 +870,9 @@ def install_dcgm_libraries(dcgm_version, target_machine): # Install DCGM. Steps from https://developer.nvidia.com/dcgm#Downloads RUN dnf config-manager --add-repo https://developer.download.nvidia.com/compute/cuda/repos/rhel8/x86_64/cuda-rhel8.repo \\ && dnf clean expire-cache \\ - && dnf install --assumeyes datacenter-gpu-manager-4-core=1:{} \\ - && dnf install --assumeyes datacenter-gpu-manager-4-devel=1:{} + && dnf install --assumeyes \\ + datacenter-gpu-manager-4-core=1:{} \\ + datacenter-gpu-manager-4-devel=1:{} """.format( dcgm_version, dcgm_version, dcgm_version ) @@ -884,8 +886,9 @@ def install_dcgm_libraries(dcgm_version, target_machine): && apt install /tmp/cuda-keyring.deb \\ && rm /tmp/cuda-keyring.deb \\ && apt update \\ - && apt install --yes datacenter-gpu-manager-4-core=1:{} \\ - && apt install --yes datacenter-gpu-manager-4-dev=1:{} + && apt install --yes --no-install-recommends \\ + datacenter-gpu-manager-4-core=1:{} \\ + datacenter-gpu-manager-4-dev=1:{} """.format( dcgm_version, dcgm_version, dcgm_version ) @@ -898,8 +901,9 @@ def install_dcgm_libraries(dcgm_version, target_machine): && apt install /tmp/cuda-keyring.deb \\ && rm /tmp/cuda-keyring.deb \\ && apt update \\ - && apt install --yes datacenter-gpu-manager-4-core=1:{} \\ - && apt install --yes datacenter-gpu-manager-4-dev=1:{} + && apt install --yes --no-install-recommends \\ + datacenter-gpu-manager-4-core=1:{} \\ + datacenter-gpu-manager-4-dev=1:{} """.format( dcgm_version, dcgm_version, dcgm_version )