search

The-Learning-And-Vision-Atelier-LAVA / DASR

[CVPR 2021] Unsupervised Degradation Representation Learning for Blind Super-Resolution
MIT License
388 stars 50 forks source link

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

Open 15937612927 opened 2 years ago

15937612927 commented 2 years ago

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 commented 2 years ago

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

15937612927 commented 2 years ago

@LongguangWang 感谢回复！ image

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