したがって、Session config proto には device_filters オプションがあり、次のコメントがあります。
// When any filters are present sessions will ignore all devices which do not
// match the filters. Each filter can be partially specified, e.g. "/job:ps"
// "/job:worker/replica:3", etc.
誰かがフォーマットの具体的な説明を持っていますか? たとえば、/gpu:0 は他のモデルを実行するために使用するため、オプションから除外したいと考えています。
私はもう試した
config = tf.ConfigProto()
config.device_filters.append('/gpu:1')
config.device_filters.append('/cpu:0')
with tf.Session(config=config):
# Do stuff
しかし、まだ ops が gpu 0 に割り当てられています。ops ごとにデバイスをオーバーライドしたくありません。