RuntimeError: Tensor for ‘out‘ is on CPU, Tensor for argument #1 ‘self‘ is on CPU, but expected them

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

产生这类错误的原因,模型和数据未在相同设备上
解决方法:
将其放在同一设备上,模型和数据都放在GPU上 net .to(“cuda”) data.to(“cuda”)

最后如果还出现相同错误,请检查网络模型定义部分,将nn.Linear() nn.LayerNorm()等放在def __init__函数里初始化
对于将不同结构组成一个module的情况,初始化时不要使用[],一定要使用nn.ModuleList()
如:

self.mlp_blocks = nn.ModuleList()
for _ in range(num_blocks):self.mlp_blocks.append(MixerBlock(tokens_mlp_dim, channels_mlp_dim, tokens_hidden_dim, channels_hidden_dim))

https://stackoverflow.com/questions/69872820/runtimeerror-tensor-for-out-is-on-cpu-tensor-for-argument-1-self-is-on-cp

https://blog.csdn.net/weixin_43760844/article/details/116047427

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

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 ...
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数据 ...
RuntimeError: set_sizes_contiguous is not allowed on a Tensor created from .data or .detach().
报错信息如下: RuntimeError Traceback (most recent call last) ~\AppData\Local\Temp/ipykernel_19156/27953557 ...
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 ...
【报错记录】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 ...
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]修改后: ...

RuntimeError: Tensor for ‘out‘ is on CPU, Tensor for argument #1 ‘self‘ is on CPU, but expected them

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

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

最新文章

热门文章