神經網絡訓練採用gpu設置

    在定義圖結構以前不用加入gpu:0,只有在session中計算以前在加入,不然的話會提示不能使用gpu保存模型等問題。session with tf.device( '/gpu:0' ):
相關文章
相關標籤/搜索