关于GPU训练时出现的错误(请大佬帮忙解析一下,跪谢)
yimu8888 • 4529 次浏览 • 0 个回复 • 2020年11月26日

下面是运行时的错误 大概意思就是训练数据在cpu,网络权重在GPU,但是要求其二者都应该在GPU
training on cuda
Traceback (most recent call last):
File "D:/python1/VGG5_7/GOOGLENET5_9/GOOGLENET5_9.py", line 73, in
d2l.train_ch5(net,train_iter,test_iter,batch_size,optimizer,device,num_epochs)
File "C:\Users\86156\Anaconda3\lib\site-packages\d2lzh_pytorch\utils.py", line 243, in train_ch5
test_acc = evaluate_accuracy(test_iter, net)
File "C:\Users\86156\Anaconda3\lib\site-packages\d2lzh_pytorch\utils.py", line 792, in evaluate_accuracy
acc_sum += (net(X).argmax(dim=1)==y).float().sum().item()
File "C:\Users\86156\Anaconda3\lib\site-packages\torch\nn\modules\module.py", line 722, in _call_impl
result = self.forward(*input, **kwargs)
File "C:\Users\86156\Anaconda3\lib\site-packages\torch\nn\modules\container.py", line 117, in forward
input = module(input)
File "C:\Users\86156\Anaconda3\lib\site-packages\torch\nn\modules\module.py", line 722, in _call_impl
result = self.forward(*input, **kwargs)
File "C:\Users\86156\Anaconda3\lib\site-packages\torch\nn\modules\container.py", line 117, in forward
input = module(input)
File "C:\Users\86156\Anaconda3\lib\site-packages\torch\nn\modules\module.py", line 722, in _call_impl
result = self.forward(*input, **kwargs)
File "C:\Users\86156\Anaconda3\lib\site-packages\torch\nn\modules\conv.py", line 419, in forward
return self._conv_forward(input, self.weight)
File "C:\Users\86156\Anaconda3\lib\site-packages\torch\nn\modules\conv.py", line 416, in _conv_forward
self.padding, self.dilation, self.groups)
RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same
原创文章,转载请注明 :关于GPU训练时出现的错误(请大佬帮忙解析一下,跪谢) - pytorch中文网
原文出处: https://ptorch.com/news/255.html
问题交流群 :168117787
- 没有评论