From 9ae913b1164690c800303ec01f52edf6cfaa4ec0 Mon Sep 17 00:00:00 2001 From: Simon Deziel Date: Sun, 17 Dec 2023 14:56:20 -0500 Subject: [PATCH] tests/gpu-container: install recommended driver Signed-off-by: Simon Deziel --- tests/gpu-container | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/gpu-container b/tests/gpu-container index 627f37d81..fd06d8646 100755 --- a/tests/gpu-container +++ b/tests/gpu-container @@ -26,7 +26,9 @@ if mokutil --sb-state | grep -Fx "SecureBoot enabled"; then fi # Install dependencies -INSTALL_RECOMMENDS=yes install_deps jq nvidia-utils-525 nvidia-driver-525 +install_deps jq ubuntu-drivers-common +RECOMMENDED_DRIVER="$(ubuntu-drivers devices 2>/dev/null | awk '/nvidia-driver-.*recommended$/ {print $3}')" +INSTALL_RECOMMENDS=yes install_deps "${RECOMMENDED_DRIVER}" # Install LXD install_lxd