7
だからセッション設定プロトはコメントで、device_filtersオプションがあります。TensorFlowのデバイスフィルタのフォーマットは何ですか?
// 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
を試みた。しかし、私はまだ私があたり-OP基づいてデバイスを無効にしたくない、GPU 0に割り当てられたOPSを取得しています。
:あなたは
/gpu:1
と/cpu:0
上でOPSを実行しているのと同じ目的を達成したい場合は、「ソフトな配置」を使用して、次のようにそれを行うことができますか?一般的な使用事例は、マルチGPU設定(つまり 'train 'と' eval 'ジョブ)のGPUの一つにすべてを割り当てることです – MZHmこれはDistributed TensorFlow用に実装されているので、' tf.train .Server'を実行してから 'tf.Session'をそのサーバに接続すると、このオプションを使うことができます。代わりに、['tf.ConfigProto.gpu_options.visible_device_list'](https://github.com/tensorflow/tensorflow/blob/4b136d9fc4de0a53c17e336965f8884fd36168a8/tensorflow/core/protobuf/config.proto#L42)オプションを使用して'tf.Session'は(ローカル)GPUのサブセットのみを使用します。 – mrry