Skip to content

Commit 7e32d54

Browse files
committed
feat: check if nvidia has nvidia stuff installed
1 parent 3cb9acd commit 7e32d54

1 file changed

Lines changed: 12 additions & 0 deletions

File tree

build_files/base/20-tests.sh

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -57,6 +57,18 @@ for package in "${UNWANTED_PACKAGES[@]}"; do
5757
echo "Unwanted package found: ${package}... Exiting"; exit 1
5858
fi
5959
done
60+
61+
if [[ "${IMAGE_NAME}" =~ nvidia ]]; then
62+
NV_PACKAGES=(
63+
libnvidia-container-tools
64+
kmod-nvidia
65+
nvidia-driver-cuda
66+
)
67+
for package in "${NV_PACKAGES[@]}"; do
68+
rpm -q "${package}" >/dev/null || { echo "Missing NVIDIA package: ${package}... Exiting"; exit 1 ; }
69+
done
70+
fi
71+
6072
IMPORTANT_UNITS=(
6173
brew-update.timer
6274
brew-upgrade.timer

0 commit comments

Comments
 (0)