Skip to content

Commit bf6a602

Browse files
committed
Install DCGM dev package
1 parent aa6fdc0 commit bf6a602

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

build.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -868,7 +868,7 @@ def install_dcgm_libraries(dcgm_version):
868868
RUN ARCH=$( [ $(uname -m) = "x86_64" ] && echo "$(uname -m)" || echo "sbsa" ) && \\
869869
&& dnf config-manager --add-repo https://developer.download.nvidia.com/compute/cuda/repos/rhel8/${{ARCH}}/cuda-rhel8.repo \\
870870
&& dnf clean expire-cache \\
871-
&& dnf install -y datacenter-gpu-manager-{}-cuda${{CUDA_VERSION%%\.*}}
871+
&& dnf install -y datacenter-gpu-manager-{}-dev
872872
""".format(
873873
dcgm_version, dcgm_version
874874
)
@@ -885,7 +885,7 @@ def install_dcgm_libraries(dcgm_version):
885885
&& apt install /tmp/cuda-keyring.deb \\
886886
&& rm /tmp/cuda-keyring.deb \\
887887
&& apt-get update \\
888-
&& apt-get install -y datacenter-gpu-manager-{}-cuda${{CUDA_VERSION%%\.*}}
888+
&& apt-get install -y datacenter-gpu-manager-{}-dev
889889
""".format(
890890
dcgm_version, dcgm_version
891891
)

0 commit comments

Comments
 (0)