Ubuntu 上如何知道程序是在 GPU 还是 CPU 上执行？

Question

我正在使用 Ubuntu 16.04，并运行我的程序来训练深度学习模型。纪元数看起来很大，但程序太慢了。我想确保我的程序在 GPU 上运行？

我怎么知道呢？

谢谢！

Answer 1

我假设您使用的是 Nvidia GPU。

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

观看 -n 1 nvidia-smi

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

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