日々精進

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

TensorFlowが使うGPUを指定する

export CUDA_VISIBLE_DEVICES="0" のように環境変数を設定するとTensorFlowが使うGPUを指定出来る。 番号は0から順に割り振られる。グリッドサーチをするときに各学習にGPUを一つずつ割り当てて並列化する場合などに使う。

参考:

qiita.com