From aaad50e147b4548d27b81da23ae46d42f6dc163d Mon Sep 17 00:00:00 2001 From: Carlos Eduardo Arango Gutierrez Date: Wed, 22 Jan 2025 11:52:58 +0100 Subject: [PATCH] Run nvidia-smi after installing the nv-drivers Signed-off-by: Carlos Eduardo Arango Gutierrez --- pkg/provisioner/templates/nv-driver.go | 2 ++ 1 file changed, 2 insertions(+) diff --git a/pkg/provisioner/templates/nv-driver.go b/pkg/provisioner/templates/nv-driver.go index dbcec586..02d89e70 100644 --- a/pkg/provisioner/templates/nv-driver.go +++ b/pkg/provisioner/templates/nv-driver.go @@ -35,6 +35,8 @@ sudo dpkg -i cuda-keyring_1.1-1_all.deb with_retry 3 10s sudo apt-get update install_packages_with_retry cuda-drivers + +nvidia-smi -L ` type NvDriver struct {