tensorflow 多GPU並行訓練 遇到的問題

os.environ['CUDA_VISIBLE_DEVICES'] = "5,6" ##如下，將graph部署到兩個GPU的時候i分別是5和6 with tf.device('/gpu:%d' % i): with tf.name_scope('tower_%d' % i) as scope: 在訓練的時候發現卡5和卡6佔的顯存相差巨大。 web 緣由 實際上os.en

>>阅读原文<<