直接在 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是关闭。
公告 : 本站提供的源码、模板、插件等等其他资源,不包含技术服务请大家谅解。赞助本站VIP会员免费提供技术服务以及远程协助 。
说明: 教程版权归原作者所有,本站只是负责搜集整理,不承担任何技术及版权问题。在本站下载任何教程仅提供学习参考,不得用于商业用途,请在下载后在24小时内删除。
须知: 本网站资源来源于网络收集,如有侵权,请联系站长进行删除处理。分享目的仅供大家学习和交流,请不要用于商业用途,否则后果自负。
声明:本站提供的教程(文章)均为网上搜集或本站原创,如果该程序涉及或侵害到您的版权可联系我们进行处理。