我怀疑是我的代码问题,因为我其他的函数中并未使用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的状态:
希望能有大佬指点一下!!!
您好,请问问题解决了吗?如已解决,能否分享一下?