浏览代码

add nvidia gpu disabler script

Noah Vogt 1 年之前
父节点
当前提交
1647760aab
共有 1 个文件被更改,包括 9 次插入0 次删除
  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