训练时出现一个错误AssertionError: Invalid device id

[15937612927]

Traceback (most recent call last): File "main.py", line 15, in model = model.Model(args, checkpoint) File "C:\Users\Luffy\Desktop\DASR-main\model_init_.py", line 29, in init self.model = nn.DataParallel(self.model, range(args.n_GPUs)) File "C:\Users\Luffy\Anaconda3\lib\site-packages\torch\nn\parallel\data_parallel.py", line 142, in init _check_balance(self.device_ids) File "C:\Users\Luffy\Anaconda3\lib\site-packages\torch\nn\parallel\data_parallel.py", line 23, in _check_balance dev_props = _get_devices_properties(device_ids) File "C:\Users\Luffy\Anaconda3\lib\site-packages\torch_utils.py", line 455, in _get_devices_properties return [_get_device_attr(lambda m: m.get_device_properties(i)) for i in device_ids] File "C:\Users\Luffy\Anaconda3\lib\site-packages\torch_utils.py", line 455, in return [_get_device_attr(lambda m: m.get_device_properties(i)) for i in device_ids] File "C:\Users\Luffy\Anaconda3\lib\site-packages\torch_utils.py", line 438, in _get_device_attr return get_member(torch.cuda) File "C:\Users\Luffy\Anaconda3\lib\site-packages\torch_utils.py", line 455, in return [get_device_attr(lambda m: m.get_device_properties(i)) for i in device_ids] File "C:\Users\Luffy\Anaconda3\lib\site-packages\torch\cuda_init.py", line 312, in get_device_properties raise AssertionError("Invalid device id") AssertionError: Invalid device id 请问是GPU设置的问题吗？

[LongguangWang]

Hi @15937612927, option.py中的n_GPUs需要与实际电脑中的显卡数量匹配。

[15937612927]

@LongguangWang 感谢回复！

我在训练时出现了如上问题，能帮我看看是什么问题吗？