Tensorflowで書いたプログラムよりGPUの利用方法？

Question

他人がTensorflowで書いたプログラム(Pythonコード)を利用したいのですが、
GPU対応をしているかどうかどうやって分かるのでしょうか。
(PCにNvidiaのGPUを３台搭載しています。)
もしかして、Tensorflowが自動的にGPUを検知し、GPUでコードを実行するのでしょうか。

もし上記のような事が成り立たなければ、どうやって、GPUを利用するのでしょうか。

どうぞ宜しくお願い致します。

Comments

[tiitoi]

pip install tensorflow-gpu をインストールしていれば GPU が使われます。

[OOZAWA]

『pip install tensorflow-gpu をインストールしていれば 』そうですか、知らなかった！　有難うございます。
ただ、これは何を意味するのでしょうか、要は「-gpu」スイッチがなければ、インストールしたのはCPU版のtensorflowのみと言うのでしょうか。
そして、GPU版のtensorflowをインストールすれば、デフォルトとして、可能な限りGPUでコードを実行してくれるという事でしょうか。

[tiitoi]

スイッチではなく、`tensorflow-gpu` という名前のパッケージです。
CPU 版 ('tensorflow')と GPU 版 ('tensorflow-gpu')はバイナリがわかれて配布されています。

> インストールしたのはCPU版のtensorflowのみと言うのでしょうか。
はい、CPU版をインストールした場合は GPU が PC に刺さっていたとしても使われません。以下の公式ガイドを参照ください。

https://www.tensorflow.org/install/gpu

[OOZAWA]

ありがとうございました。

Answer 1

tensorflow 1.15あたりまでは定義したグラフを任意のデバイスに割り当て可能でした。デフォルトは決まっているかと思いますが、私はGPUメモリの割り当てる、タスクごとにGPUを割り当てるといった調整目的もあり、コードに明示的に記載しています。

Comments

[OOZAWA]

ご教授ありがとうございます！
『定義したグラフを任意のデバイスに割り当て可能でした。、、、コードに明示的に記載しています。』　自由度結構高いですね。　この辺勉強しなければなりません。

Answer 2

こちら参考にして確認してみては？
TensorFlow-GPUのインストールで詰まったときのメモ

その他いろいろ （ググるだけでいっぱいヒットしますよ。）
tensorflow gpu 確認

Comments

[OOZAWA]

貴重な情報有難うございます！
助かります。