日々精進

新しく学んだことを書き留めていきます

2016-10-28から1日間の記事一覧

Kerasで複数GPUを使って学習する

AI

何もしなくても全GPUを使って学習してくれるかと思ってたけどそうではなかった。 以下のようにしてGPU数を渡してやると全GPUを使ってくれる。 def get_available_gpus(): local_device_protos = device_lib.list_local_devices() return [x.name for x in lo…