只需在参数选择项增加一项gpu调用模块。code 如下:
parser.add_argument('--gpu-id', default='0', type=str,
help='id(s) for CUDA_VISIBLE_DEVICES')
只需在参数选择项增加一项gpu调用模块。code 如下:
parser.add_argument('--gpu-id', default='0', type=str,
help='id(s) for CUDA_VISIBLE_DEVICES')