我要确认我正在使用具有张量流的GPU,但我有点困惑。所以,我使用以下代码来检查是否使用了GPU:
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
输出以下内容:
根据我的理解,从输出中可以看出GPU用于计算。但是当我用它监视它时:
~$sudo nvidia-smi --loop=1
我有以下输出,它不断给出以下结果:
从我在上一张照片中的未成年人来看,它并没有使用GPU。这是我想念的东西吗?有提示吗?
编辑(在Yaroslav Bulatov发表评论之后,这有很多帮助)
我现在可以确认我正在使用GPU。对于matmul(A,B)
,我从ctf文件中获得了以下结果:
然而,tensorflow中有一些函数似乎正在使用CPU。例如tf.nn.top_k
:
有趣..