Shichao
Shichao
存在一个bug,当传入参数`device`并非从0开始(),例如传入2,3或者1,2 时候,提示报错:AssertionError: Invalid device id。 - 这个bug产生的原因是当75行代码设置了全局GPU的数量,如下: `os.environ["CUDA_VISIBLE_DEVICES"] = args.device # 此处设置程序使用哪些显卡` - 而多GPU平行模型的代码如下: `model = DataParallel(model, device_ids=[int(i) for i in args.device.split(',')])` 问题在于device_ids读入的id是实际id号而因为设置了环境变量CUDA_VISIBLE_DEVICES,导致gpu 的id不一致。 以传入两个gpu id为例: 1. 如果传入参数是`0,1` 环境变量识别出两个GPU,则GPU的id为`0,1`,工作正常; 2....
Hi, thank you for this nice tool and also thanks for providing a windows version. I have to work on a sever with Windows Server 8 R2 and I had...