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

Question

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

我的模型正在运行时检查了nvidia-smi，我发现python被识别为一个进程但是用法是N / A.

我还尝试运行caffe.set_mode_cpu（）命令，认为时间会有很大的不同，但是命令的时间和没有相同的时间。

Answer 1

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