直接在 go 文件中添加命令
通过脚本设置
#!/bin/bash # 检查驱动是否安装 command -v nvidia-smi &> /dev/null || { echo >&2 "nvidia driver is not installed you will need to install this from community applications... exiting."; exit 1; } echo "本机已安装Nvidia驱动程序" echo echo "在当前的服务器上找到以Nvidia显卡" echo nvidia-smi --list-gpus echo echo "-------------------------------------------------------------" # 为显卡设置持久性模式 nvidia-smi --persistence-mode=1 # 查询电源状态 gpu_pstate=$(nvidia-smi --query-gpu="pstate" --format=csv,noheader); # 查询使用 GPU 的进程的 PID gpupid=$(nvidia-smi --query-compute-apps="pid" --format=csv,noheader); # 通过检查字符串中是否存在任何 PID 来检查 PState 是否为零并且没有运行的进程。 if ( "$gpu_pstate" == "P0" ) && ( -z "$gpupid" ); then echo "没有找到对应 PID,因此没有正在运行的进程" fuser -kv /dev/nvidia* echo "电源状态为:" echo "$gpu_pstate" # 显示当前电源状态 else echo "电源状态为:" echo "$gpu_pstate" # 显示当前电源状态 fi echo echo "-------------------------------------------------------------" echo echo "当前功耗为:" # 检查GPU的当前功耗 nvidia-smi --query-gpu=power.draw --format=csv exit
- 设置脚本运行时机:保存脚本后,在右边选择 “At Starup of Array”,使脚本在阵列启动时运行
相关注意事项
PS:大家自行测试是关闭的好,还是开启的好。国际惯例nvidia-smi --persistence-mode=1是开启,nvidia-smi --persistence-mode=0是关闭。
声明:
1.本站为个人非盈利站点,旨在个人学习、欣赏及记录等,故不受狭义的商业性版权限制,除非特别声明;
2.本站主要内容来源为本站编辑撰写、网友投稿(包括原创及非原创)、翻译外文和转载其他网站。