直接在 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小时内删除。
须知: 本网站资源来源于网络收集,如有侵权,请联系站长进行删除处理。分享目的仅供大家学习和交流,请不要用于商业用途,否则后果自负。
声明:本站提供的教程(文章)均为网上搜集或本站原创,如果该程序涉及或侵害到您的版权可联系我们进行处理。