ci: Anchor datacenter-gpu-manager-4-core version (#8315)

Co-authored-by: Misha Chornyi <[email protected]>

Author: yinggeh

Dockerfile.sdk

@@ -265,7 +265,10 @@ 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 --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).

build.py

@@ -858,19 +858,23 @@ 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-devel=1:{}
+    && dnf install --assumeyes \\
+                 datacenter-gpu-manager-4-core=1:{} \\
+                 datacenter-gpu-manager-4-devel=1:{}
 """.format(
-                    dcgm_version, dcgm_version
+                    dcgm_version, dcgm_version, dcgm_version
                 )
             else:
                 return """
 ENV DCGM_VERSION {}
 # 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-devel=1:{}
+    && dnf install --assumeyes \\
+                 datacenter-gpu-manager-4-core=1:{} \\
+                 datacenter-gpu-manager-4-devel=1:{}
 """.format(
-                    dcgm_version, dcgm_version
+                    dcgm_version, dcgm_version, dcgm_version
                 )
         else:
             if target_machine == "aarch64":
@@ -882,9 +886,11 @@ 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-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, dcgm_version, dcgm_version
                 )
             else:
                 return """
@@ -895,9 +901,11 @@ 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-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, dcgm_version, dcgm_version
                 )