如何判断自己是在使用cpu还是gpu跑代码

如何判断自己是在使用cpu还是gpu跑代码

在Linux系统下，可以通过在终端中输入一句命令来查看自己跑的代码是在用CPU还是GPU：

nvidia-smi

举例：

上图可以看出，GPU利用率为5%，当前代码是在使用cpu。

上图可以看出，GPU利用率为83%，当前代码是在使用gpu。

除此以外，使用该句命令也可以查看自己的gpu版本和cuda版本。

具体每个参数代表什么意思，可以通过参考这篇文章https://blog.csdn.net/a784586/article/details/78688842