NVIDIA 显卡限制功率

1.应用背景

给实验室的一台服务器Dell T7910加装了两块3090显卡，但是由于供电口不够，所有用了两个6pin转8pin的转接线进行转接，导致显卡的电源输入功率不够，跑实验的时候一跑到满载就会扛不住关机重启，暂时无法更换电源，因此考虑对显卡功率进行限制。

2.配置实现

2.1 临时设置

nvidia-smi -pm 1 # enable persistance mode
nvidia-smi -pl 250 # set power limit to 250W

• 3090配置前的最大功率为350W，因供电原因配置限制后的最大功率为250W

2.2 永久设置

1. 新建nvidia-setpower.service文件

   sudo vim /etc/systemd/system/nvidia-setpower.service

   [Unit]
   Description=Nvidia SetPower Service
   After=network.target
   Wants=network.target
   
   [Service]
   Type=simple
   PIDFile=/run/nvidia-setpower.pid
   ExecStart=sh /usr/bin/nvidia-setpower.sh
   Restart=on-failure
   # Don't restart in the case of configuration error
   RestartPreventExitStatus=23
   
   [Install]
   WantedBy=multi-user.target

2. 新建nvidia-setpower.sh

   sudo vim /usr/bin/nvidia-setpower.sh

   sudo nvidia-smi -pm 1 
   sudo nvidia-smi -pl 250

   sudo nvidia-smi -i 1 -pl 250，-i可以指定第几个显卡。

3. 设置服务开机启动：

   systemctl daemon-reload
   systemctl start nvidia-setpower.service
   systemctl enable nvidia-setpower.service

参考资料

1. NVIDIA 显卡在 Ubuntu下限制其功率的方法
2. ubuntu 永久 设置 降低N显卡功率 nvidia-smi