似乎tf.train.replica_device_setter不允许指定使用的gpu.
我想做的是如下:
with tf.device(
tf.train.replica_device_setter(
worker_device='/job:worker:task:%d/gpu:%d' % (deviceindex, gpuindex)):
<build-some-tf-graph>