Warning: Found GPU0 GeForce GT 720M which is of cuda capability 2.1. PyTorch no longer supports this GPU because it is too old. The minimum cuda capability that we support is 3.5. warnings.warn(old_gpu_warn % (d, name, major, capability[1]))
0%| | 0/938 [00:00<?, ?step/s]
--------------------------------------------------------------------------- RuntimeError Traceback (most recent call last) <ipython-input-13-59a3d3b0c6bb> in <module> 8 9 net.zero_grad() ---> 10 outputs = net(train_imgs) 11 loss = loss_fun(outputs,labels) 12 predictions = torch.argmax(outputs, dim = 1) ~\Anaconda3\envs\pytorch\lib\site-packages\torch\nn\modules\module.py in __call__(self, *input, **kwargs) 491 result = self._slow_forward(*input, **kwargs) 492 else: --> 493 result = self.forward(*input, **kwargs) 494 for hook in self._forward_hooks.values(): 495 hook_result = hook(self, input, result) <ipython-input-9-c306382fe6a2> in forward(self, input) 26 27 def forward(self,input): ---> 28 output = self.model(input) 29 return output ~\Anaconda3\envs\pytorch\lib\site-packages\torch\nn\modules\module.py in __call__(self, *input, **kwargs) 491 result = self._slow_forward(*input, **kwargs) 492 else: --> 493 result = self.forward(*input, **kwargs) 494 for hook in self._forward_hooks.values(): 495 hook_result = hook(self, input, result) ~\Anaconda3\envs\pytorch\lib\site-packages\torch\nn\modules\container.py in forward(self, input) 90 def forward(self, input): 91 for module in self._modules.values(): ---> 92 input = module(input) 93 return input 94 ~\Anaconda3\envs\pytorch\lib\site-packages\torch\nn\modules\module.py in __call__(self, *input, **kwargs) 491 result = self._slow_forward(*input, **kwargs) 492 else: --> 493 result = self.forward(*input, **kwargs) 494 for hook in self._forward_hooks.values(): 495 hook_result = hook(self, input, result) ~\Anaconda3\envs\pytorch\lib\site-packages\torch\nn\modules\conv.py in forward(self, input) 336 _pair(0), self.dilation, self.groups) 337 return F.conv2d(input, self.weight, self.bias, self.stride, --> 338 self.padding, self.dilation, self.groups) 339 340 RuntimeError: cuDNN error: CUDNN_STATUS_ARCH_MISMATCH
简单点说就是GPU能力不行,我暂时没有找到好办法去处理,有没有大佬可以救救