elektronn2.utils.gpu module

elektronn2.utils.gpu.get_free_gpu(wait=0, nb_gpus=-1)[source]
elektronn2.utils.gpu.initgpu(gpu)[source]