神经网络训练采用gpu设置

    在定义图结构以前不用加入gpu:0,只有在session中计算以前在加入,不然的话会提示不能使用gpu保存模型等问题。session with tf.device( '/gpu:0' ):
相关文章
相关标签/搜索