8

したがって、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 ごとにデバイスをオーバーライドしたくありません。

4

1 に答える 1