tests/gpu-container: install recommended driver

Signed-off-by: Simon Deziel <[email protected]>
canonical · Dec 17, 2023 · 9ae913b · 9ae913b
1 parent 1e2283a
commit 9ae913b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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