LazyYangHuan 2019-05-22 20:23 采纳率: 0%
浏览 4545

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

这是主函数部分

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'):
            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)

下面是运行时CPU/GPU状态:

图片说明

Pycharm中的显示:
图片说明

我怀疑是我的代码问题,因为我其他的函数中并未使用tensorflow的操作,会不会未使用tf中的命令,所以不能让gpu跑程序?

求大神解答

  • 写回答

2条回答 默认 最新

  • threenewbee 2019-05-22 21:10
    关注

    你打开任务管理器的gpu界面看下
    如果你看到gpu内存分配了,gpu内存传输的占用率上来了,那么就ok
    我发现一些情况下,gm/gp系列的gpu(也就是9代、10代)会显示不出计算负载。

    评论

报告相同问题?