这是主函数部分
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状态:
我怀疑是我的代码问题,因为我其他的函数中并未使用tensorflow的操作,会不会未使用tf中的命令,所以不能让gpu跑程序?
求大神解答
你打开任务管理器的gpu界面看下
如果你看到gpu内存分配了,gpu内存传输的占用率上来了,那么就ok
我发现一些情况下,gm/gp系列的gpu(也就是9代、10代)会显示不出计算负载。
请问你的问题解决了吗 我也遇到了