在定义图结构之前不用加入gpu：0，只有在session中计算之前在加入，否则的话会提示不能使用gpu保存模型等问题。
with
tf.device(
'/gpu:0'
):
神经网络训练采用gpu设置
猜你喜欢
转载自blog.csdn.net/weixin_41863859/article/details/80222361
今日推荐
周排行