複数の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 登録するときは、最低限この辺きちんと作らなきゃなー