tensorflow lib中の関数が自動的にGPUによって実行してくれる？

Question

tensorflowを利用処して 深層機械学習を実装する場合、もしPCにCUDA　GPUが装着していれば、
どうやって、コードの実行をGPUで行ってくれるのでしょうか。

例えば、何らかの指示文を少々入れれば済ませる程度ですか、
それとも、GPUに対応するためにかなりの実装を自分でやらなければならないのでしょうか。

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

Answer 1

tensorflowのインストール時に使うライブラリを切り替えることでGPUに対応することができます。

#pip install --upgrade tensorflow-gpu

詳しくはUbuntuなどでのインストールの例などを見てください。
https://www.tensorflow.org/install/install_linux

基本的に、ソースコード上でGPUに対応するためのコーディングは不要で、CPUのみの場合と同じコードで動きます。

Comments

[oookabe]

情報有難うございます。
素晴らしいです。