train.py 文件源码

def gpu_conf(cfg, gpu_id=None):

            if gpu_id==None:
                DEVICE_ID_LIST = GPUtil.getFirstAvailable()
                if (len(DEVICE_ID_LIST) > 0):
                    cfg.GPU_ID = DEVICE_ID_LIST[0]  # grab first element from list
            else:
                cfg.GPU_ID=gpu_id

            return cfg