我在tensorflow中指定了GPU运行代码,但是为什么只有CPU在满速工作,GPU没变化呢?

新手上路,请多包涵

我怀疑是我的代码问题,因为我其他的函数中并未使用tensorflow的操作,会不会是因为不使用tf中的命令,所以不能让gpu跑程序?
我是在windows下运行的(tensorflow-gpu 1.12.0)
这是我主函数:

if __name__=="__main__":
    with tf.Session(config=tf.ConfigProto(allow_soft_placement=True, log_device_placement=True)) as sess:
        with tf.device('/device:GPU:0'):
            ###后面是跑LDA线性判别的代码
            n,m=x.shape
            print(n,m)
            X=np.zeros((n//72,72,m))
            for i in range(n//72):
                xi=x[i*72:i*72+72,:]
                X[i]=xi
            Showimage(X,x,n//72,m)

这是运行的时候的GPU-CPU状态:
图片描述

这是Pycharm的状态:
图片描述

希望能有大佬指点一下!!!

阅读 6.2k
1 个回答
撰写回答
你尚未登录,登录后可以
  • 和开发者交流问题的细节
  • 关注并接收问题和回答的更新提醒
  • 参与内容的编辑和改进,让解决方法与时俱进
推荐问题