(args)
| 29 | |
| 30 | |
| 31 | def set_gpu(args): |
| 32 | gpu_list = [int(x) for x in args.gpu.split(',')] |
| 33 | print('use gpu:', gpu_list) |
| 34 | os.environ['CUDA_DEVICE_ORDER'] = 'PCI_BUS_ID' |
| 35 | os.environ['CUDA_VISIBLE_DEVICES'] = args.gpu |
| 36 | return gpu_list.__len__() |
| 37 | |
| 38 | |
| 39 | def ensure_path(path): |