我遇到的情况不是加个cpu就解决了,那个我会。
此处是
net = torch.nn.DataParallel(net.cuda())
File "D:\python\lib\site-packages\torch\nn\modules\module.py", line 305, in cuda
return self._apply(lambda t: t.cuda(device))
File "D:\python\lib\site-packages\torch\nn\modules\module.py", line 202, in _apply
module._apply(fn)
File "D:\python\lib\site-packages\torch\nn\modules\module.py", line 202, in _apply
module._apply(fn)
File "D:\python\lib\site-packages\torch\nn\modules\module.py", line 224, in _apply
param_applied = fn(param)
File "D:\python\lib\site-packages\torch\nn\modules\module.py", line 305, in <lambda>