有的时候明明在log里面显示已经成功使用gpu了,但是nvidia-smi查看发现gpu使用率不高,也没看到进程号。

一种情况就是实际上没用到gpu,另一种情况可能是gpu使用率确实不高,但有在使用。

这里首先去查看自己的代码是不是用到了gpu,比如使用pytorch的话,print(torch.cuda.get_device_name(0))看看gpu的名字,是否有真的使用到。

如果有的话,可以再使用watch -n 10 nvidia-smi来周期性显示gpu的情况,因为直接使用nvidia-smi只能看到瞬间的状态,这里10就是10s。我们周期性来看是不是有进程在使用gpu。

如果还是看不到gpu在使用的话,使用fuser -v /dev/nvidia*查看使用gpu的进程号,这里的进程都在使用gpu,但可能使用的比较少,nvidia-smi可能有的不会显示,再对比自己的进程号就能看到有没有在使用gpu啦。

如果有帮助解决到问题请帮忙顺手点个赞,能有更大的动力把解决到的问题分析给大家哈。

本文地址:https://blog.csdn.net/qq_33624866/article/details/110866874