Merge pull request #264 from ArangoGutierrez/smi

Run nvidia-smi after installing the nv-drivers
NVIDIA · Jan 22, 2025 · b4801da · b4801da
2 parents 4d62389 + aaad50e
commit b4801da
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 {