File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -868,7 +868,7 @@ def install_dcgm_libraries(dcgm_version):
868868RUN 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 )
You can’t perform that action at this time.
0 commit comments