|
发表于 2022-7-18 20:34:23
|
显示全部楼层
在main.py中找到选择gpu的代码,替换force_gpu_idxs = arguments.force_gpu_idxs,
为force_gpu_idxs = [ int(x) for x in arguments.force_gpu_idxs.split(',') ] if arguments.force_gpu_idxs is not None else None,
下面选择参数的语句p.add_argument('--force-gpu-idxs', dest="force_gpu_idxs", default=None, help="Force to choose GPU indexes separated by comma.")
替换为:p.add_argument('--force-gpu-idxs', type=str, dest="force_gpu_idxs", default='0', help="Force to choose GPU indexes separated by comma.")
就可以啦 |
|