Tensor for ‘out‘ is on CPU, Tensor for argument #1 ‘self‘ is on CPU

1、问题
模型训练完后进行测试，报错

RuntimeError: Tensor for 'out' is on CPU, Tensor for argument #1 'self' is on CPU, but expected them to be on GPU (while checking arguments for addmm)

2、原因
将模型送入GPU之后才加载之前训练好的模型，导致模型加载到了GPU，然而你的网络权重还在CPU中，此时会报错如下，报错部分代码：

device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
net.to(device)
net.load_state_dict(torch.load('./results/bestmodel30.pth'),False)

3、解决
换下位置即可


net.load_state_dict(torch.load('./results/bestmodel30.pth'),False)
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
net.to(device)

原文链接：https://blog.csdn.net/weixin_43760844/article/details/116047427

Tensor for ‘out‘ is on CPU, Tensor for argument #1 ‘self‘ is on CPU相关推荐

PyTorch 笔记（03）— Tensor 数据类型分类（默认数据类型、CPU tensor、GPU tensor、CPU 和 GPU 之间的转换、数据类型之间转换）
1. Tensor 数据类型 Tensor 有不同的数据类型,如下表所示,每种类型都有 CPU 和 GPU 版本(HalfTensor)除外,默认的 tensor 是数据类型是 FloatTensor ...
06_2_Pytorch的基础数据类型、CPU tensor类型和GPU tensor类型、判断数据类型、CPU或GPU张量之间的转换、数据类型转换、1-4维向量、Tensor张量、Variable等
1.6.2.Pytorch的基础数据类型 1.6.2.1.Torch定义了的七种CPU tensor类型和八种GPU tensor类型 1.6.2.2.基础数据类型 1.6.2.3.Pytorch数据 ...
tensor判断是否相等_PyTorch的Tensor(中)
背景在PyTorch的Tensor系列上一篇文章中: Gemfield:PyTorch的Tensor(上)zhuanlan.zhihu.com Gemfield介绍了一个Tensor的创建过程,特 ...
IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python
报错原因分析: train_loss += loss.data[0] 是旧版本的Pytorch版本代码,在Pytorch0.4-0.5中会出现警告,不会报错, 在Pytorch0.5版本以上会报错 I ...
pytorch tensor 初始化_PyTorch简明笔记[1]-Tensor的初始化和基本操作
听麻麻说,偷偷收藏而不感谢是不礼貌的,至少应该点个赞~我觉得麻麻说的对! 不断地被人安利PyTorch,终于忍不住诱惑决定入坑了. 当初学习TensorFlow的时候,没有系统性地学习.之前TF的英文 ...
解决IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Pyth
解决IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Pyth ...
invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a python number
提示 Use tensor.item() to convert a 0-dim tensor to a Python number 原语句:running_loss+=loss.data[0]修改后: ...
关于对cpu的理解和kvm虚拟机到物理cpu的绑定
这段时间一直在想,云计算除了虚拟化之外,还应该有其它的东西,那就是优化.因为我们虚拟出来资源之后怎么用,怎么划分,是并行也好,是租给用户使用也好,都要实现资源调度和使用的最优化.嗯,这是这段时间关于云 ...

Tensor for ‘out‘ is on CPU, Tensor for argument #1 ‘self‘ is on CPU

Tensor for ‘out‘ is on CPU, Tensor for argument #1 ‘self‘ is on CPU相关推荐

最新文章

热门文章