Install libnvidia-ml.so.1 in test environment

This commit is contained in:
Ashwin Srinath
2025-12-03 14:56:37 -05:00
parent c2c34c9378
commit b7eaf44ca3

View File

@@ -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