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

Question

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

我怎么知道呢？

谢谢！

Answer 1

我假设您正在使用Nvidia GPU。

最简单的方法是在终端中键入以下命令：

  

watch -n 1 nvidia-smi

这将使您连续更新（每秒），而无需在终端上填充输出。

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