如何知道Ubuntu上的GPU或CPU上正在执行程序？

Question

我正在使用Ubuntu 16.04，并运行我的程序来训练DeepLearning模型。时代数似乎很大，但程序太慢了。我想确保我的程序在GPU上运行？

我怎么知道的？

谢谢！

Answer 1

我假设您使用的是Nvidia GPU。

实现此目的的最简单方法是在终端内键入以下命令：

看-n 1 nvidia-smi

这将为您提供连续更新（每秒）而无需使用输出填充终端。

您将能够检查哪些进程正在使用您的GPU，分配的内存，温度和使用的功率等。