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