Sfoglia il codice sorgente

add nvidia gpu disabler script

Noah Vogt 1 anno fa
parent
commit
1647760aab
1 ha cambiato i file con 9 aggiunte e 0 eliminazioni
  1. 9 0
      local-bin/disable-gpu

+ 9 - 0
local-bin/disable-gpu

@@ -0,0 +1,9 @@
+#!/bin/sh
+
+gpu_num="$(sudo nvidia-smi | grep 00: | sed 's/|//; s/^.*00://; s/:.*$//')"
+
+sudo nvidia-smi -i 0000:"$gpu_num":00.0 -pm 0
+sudo nvidia-smi drain -p 0000:"$gpu_num":00.0 -m 1
+if ! nvidia-smi | grep -q 00: ; then
+    sudo systemctl suspend
+fi