Tensorflowで使用するGPUはどれですか？

Question

私のコンピュータには2つのGPUがインストールされています。コードを実行すると、Tensorflowのデバッグ情報が表示され、2枚のGPUカードの情報が表示されます。一方、私はTFのために指定しなかった場合に使用されるだろうか？私は複数のGPUカードを持っている場合、私はTFのための1つのGPUを効率のために指定する必要があるかどうか。ありがとう〜！

Answer 1

Tensorflowは、システム上で利用可能なすべてのGPUを使用します。

GPUが2つあり、Tensorflowが両方を見ている場合、両方のデバイスに同じメモリが割り当てられ、スケジューリングアルゴリズムを使用して必要なときに1つのGPUメモリから別のGPUメモリに要素をスワップします。

コンテキストマネージャーwith tf.device("/gpu:<number>")を使用して、指定した操作を実行するGPUを決めることができます。しかし、メモリはすべてのGPUに割り当てられています。

代わりにいくつかのGPUのみを使用し、他のGPUの割り当てを避けるには、CUDA環境変数CUDA_VISIBLE_DEVICESを使用する必要があります。

Pythonスクリプトを実行する前に、Tensorflowに表示されるCPUを指定できます。このようにして、一部のデバイスだけが使用されると確信しています。あなたはTensorflowは、あなたがそのようにスクリプトを実行することができる唯一のあなたの第二GPUを見ているしたい場合例えば

、：あなたは複数のGPUを利用したい場合は、デフォルトのtensorflowによって

CUDA_VISIBLE_DEVICES=1 python execute_script.py