确认在张量流中使用GPU

时间:2017-01-31 15:18:28

标签: tensorflow gpu

我要确认我正在使用具有张量流的GPU,但我有点困惑。所以,我使用以下代码来检查是否使用了GPU:

sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

输出以下内容:

enter image description here

根据我的理解,从输出中可以看出GPU用于计算。但是当我用它监视它时:

~$sudo nvidia-smi --loop=1

我有以下输出,它不断给出以下结果:

enter image description here

从我在上一张照片中的未成年人来看,它并没有使用GPU。这是我想念的东西吗?有提示吗?

编辑(在Yaroslav Bulatov发表评论之后,这有很多帮助)

我现在可以确认我正在使用GPU。对于matmul(A,B),我从ctf文件中获得了以下结果:

enter image description here

然而,tensorflow中有一些函数似乎正在使用CPU。例如tf.nn.top_kenter image description here

有趣..

0 个答案:

没有答案