From b7eaf44ca316101678e5bdcb7552cf9255322606 Mon Sep 17 00:00:00 2001 From: Ashwin Srinath Date: Wed, 3 Dec 2025 14:56:37 -0500 Subject: [PATCH] Install libnvidia-ml.so.1 in test environment --- ci/test_pynvbench_inner.sh | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/ci/test_pynvbench_inner.sh b/ci/test_pynvbench_inner.sh index f9c47ff..ffccb86 100755 --- a/ci/test_pynvbench_inner.sh +++ b/ci/test_pynvbench_inner.sh @@ -9,6 +9,11 @@ set -euo pipefail echo -e "#!/bin/bash\nsource /opt/rh/gcc-toolset-13/enable" >/etc/profile.d/enable_devtools.sh source /etc/profile.d/enable_devtools.sh +# Install NVIDIA Management Library (required by pynvbench at runtime) +# This is a driver library that we exclude from the wheel but need for testing +echo "Installing libnvidia-ml.so.1..." +/workspace/ci/util/retry.sh 5 30 dnf -y install nvidia-driver-libs + # Set up Python environment (only if not already available) source /workspace/ci/pyenv_helper.sh if ! command -v python${py_version} &> /dev/null; then