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

Posted 2023-03-12

【中文标题】如何检查 Caffe 是不是正在使用我的 GPU？

【英文标题】：How to check if Caffe is using my GPU?如何检查 Caffe 是否正在使用我的 GPU？

【发布时间】：2018-11-23 01:07:15

【问题描述】：

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

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

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

【参考方案1】：

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