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
相关文章
相关标签/搜索