如何检查 Caffe 是否正在使用我的 GPU？

Question

与question 相同，但用于 caffe。我想要一个可以放入我的 python 脚本的命令来检查是否使用了 gpu。

我在模型运行时检查了 nvidia-smi，发现 python 被识别为进程，但 Usage 为 N/A。

我还尝试运行 caffe.set_mode_cpu() 命令，认为时间会非常不同，但使用命令的时间和不使用命令的时间相同。

Answer 1

我想建议使用GPUSTAT。您可以查询 GPU 并检查您的进程是否在命令返回的列表中。 它很简单，不太优雅，但很有效。