多GPU的数据训练，feed

 

SRE实战 互联网时代守护先锋，助力企业售后服务体系运筹帷幄！一键直达领取阿里云限量特价优惠。 
images, labels = cifar10.distorted_inputs()
split_images = tf.split(images, FLAGS.num_gpus, 0)
split_labels = tf.split(labels, FLAGS.num_gpus, 0)

 

for i in xrange(FLAGS.num_gpus):
  with tf.device('/gpu:%d' % i):
    with tf.name_scope('%s_%d' % (cifar10.TOWER_NAME, i)) as scope:

      loss = tower_loss(scope, split_images[i], split_labels[i])

 

https://stackoverflow.com/questions/34273951/tensorflow-multi-gpu-single-input-queue?

--

 

扫码关注我们
微信号：SRE实战
拒绝背锅 运筹帷幄