1、问题
2、原因
3、解决

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)
...

解决：RuntimeError: Tensor for ‘out‘ is on CPU, Tensor for argument #1 ‘self‘ is on CPU, but expected t相关推荐

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 ...
如何解决“RuntimeError: CUDA Out of memory”问题
点击上方"AI公园",关注公众号,选择加"星标"或"置顶" 作者:Nitin Kishore 编译:ronghuaiyang 导读一些可以 ...
解决RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cp
今天在把.pt文件转ONNX文件时,遇到此错误. 报错 RuntimeError: Expected all tensors to be on the same device, but found a ...
【报错记录】RuntimeError: don‘t know how to restore data location of torch.FloatStorage (tagged with CPU)
问题描述在调试一个代码时,在输入参数都正确的情况下,报出 RuntimeError: don't know how to restore data location of torch.FloatSt ...
PyTorch 笔记（03）— Tensor 数据类型分类（默认数据类型、CPU tensor、GPU tensor、CPU 和 GPU 之间的转换、数据类型之间转换）
1. Tensor 数据类型 Tensor 有不同的数据类型,如下表所示,每种类型都有 CPU 和 GPU 版本(HalfTensor)除外,默认的 tensor 是数据类型是 FloatTensor ...
成功解决RuntimeError: Selected KDE bandwidth is 0. Cannot estiamte density
成功解决RuntimeError: Selected KDE bandwidth is 0. Cannot estiamte density 目录解决问题解决思路解决方法 1.重新定义bw参数 ...
成功解决RuntimeError: filter weights array has incorrect shape.
成功解决RuntimeError: filter weights array has incorrect shape. 目录解决问题解决思路解决方法解决问题 File "F:\Pro ...
成功解决RuntimeError: Decompression ‘SNAPPY‘ not available. Options: [‘GZIP‘, ‘UNCOMPRESSED‘]
成功解决RuntimeError: Decompression 'SNAPPY' not available. Options: ['GZIP', 'UNCOMPRESSED'] 目录解决问题解 ...
成功解决RuntimeError: Java is not installed, or the Java executable is not on system path
成功解决RuntimeError("Java is not installed, or the Java executable is not on system path") Ru ...
无盘服务器 cpu占用,顺网科技网维大师程序占CPU资源多的解决过程
顺网客科技的主打网吧产品就是网维大师了,网维大师在网吧和网咖行业用户量比较大,碰到各种奇怪的问题也是蛮多的,今天就来分享一个关于「顺网科技网维大师程序占CPU资源多」问题的解决过程,分享给大家,大家碰 ...

解决：RuntimeError: Tensor for ‘out‘ is on CPU, Tensor for argument #1 ‘self‘ is on CPU, but expected t

1、问题

2、原因

3、解决

解决：RuntimeError: Tensor for ‘out‘ is on CPU, Tensor for argument #1 ‘self‘ is on CPU, but expected t相关推荐

最新文章

热门文章