Run nvidia-smi after installing the nv-drivers

Signed-off-by: Carlos Eduardo Arango Gutierrez <eduardoa@nvidia.com>
NVIDIA · Jan 22, 2025 · aaad50e · aaad50e
1 parent b0f8bf7
commit aaad50e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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 {