diff --git a/Dockerfile.sdk b/Dockerfile.sdk index 7d114e2ce5..8fbc8cd3c0 100644 --- a/Dockerfile.sdk +++ b/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). diff --git a/build.py b/build.py index 230b4dc88c..6c726e6c87 100755 --- a/build.py +++ b/build.py @@ -858,9 +858,11 @@ 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 """ @@ -868,9 +870,11 @@ 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-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 )