gorogoroyasu

福岡の開発会社で働いている。

複数のGPUを積んだマシンで、未使用のGPU を選ぶやつ作った。

タイトルが難しい。

とりあえず、作りました。
github.com

作り方はとっても簡単。

nvidia-smi

コマンドたたいて、結果をCSV で受け取るだけ。

使い方はもっと簡単。

使い方

from get_unused_gpu.main import get_unused_gpu


os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = str(get_unused_gpu())

python のお作法は知らないので、git clone して、適当に使ってください。

PyPI 一応登録はしたものの、ちょっと怖くて使えてません。
気が向いたら登録します。

とりあえず、自分用です。
なんか、こんなブログですいません。。。

あ、ちなみに、

gpu temp: under 40°C
utilization.gpu: 0 %
utilization.memory: 0 %

が未使用と判定する条件です。
時系列情報とかとってないし、同時に処理が実行されたときのこととか関係ないのであしからず。
あと、現状使用可能なGPU がない場合 exit() してるので、止まります。
PiPY 登録するときは、最低限この辺きちんと作らなきゃなー