这是主函数部分
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跑程序?
求大神解答