chore(install-nvidia-driver): do not call nvidia-smi when the NVIDI…

…A kernel modules are not loaded
XuehaiPan · Nov 10, 2023 · 83f90f3 · 83f90f3
1 parent 36e66bb
commit 83f90f3
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/install-nvidia-driver.sh b/install-nvidia-driver.sh
@@ -460,7 +460,9 @@ for dm in "${DM_SERVICES[@]}"; do
 done
 
 # Disable persistence mode
-exec_cmd "sudo nvidia-smi -pm 0 || true"
+if [[ -n "$(sudo lsmod | grep '^nvidia' | awk '{ print $1 }')" ]]; then
+	exec_cmd "sudo nvidia-smi -pm 0 || true"
+fi
 
 sleep 1 # ensure the processes are stopped