报错：RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda

超、凢脫俗 2023-10-15 21:06 155阅读 0赞

详细的报错信息如下：

RuntimeError: Expected all tensors to be on the same device, 
but found at least two devices, cuda:0 and cpu! 
(when checking argument for argument mat1 in method wrapper_mm)

报错位置

报错的位置是：网络的forward函数(linear.py/forward)
在这里插入图片描述

解决办法

在这里插入图片描述

发表评论取消回复

表情：

评论列表（有 0 条评论，155人围观）

还没有评论，来说两句吧...

相关阅读

相关讲解All input tensors must be on the same device

目录讲解All input tensors must be on the same device 错误原因解决方法示例代码结论 --------------

悠悠/ 2024年02月05日 22:00/ 0 赞/ 143 阅读

相关报错：RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda

详细的报错信息如下： RuntimeError: Expected all tensors to be on the same device, but fo

超、凢脫俗/ 2023年10月15日 21:06/ 0 赞/ 156 阅读

相关 MyBatis报错：Expected one result (or null) to be returned by selectOne(), but found: 2

[欢迎访问我的网站:www.ifueen.com][www.ifueen.com] Mybatis报错问题一次项目在使用MyBatis进行高级查询的时候，如果传入的值

叁歲伎倆/ 2023年10月08日 09:27/ 0 赞/ 105 阅读

相关 RuntimeError:CUDA error:no kernel image is available for execution on the device报错解决(亲测)

深度学习算法训练报错调试Transformer网络，安装完timm包之后，运行程序时报错`CUDA error:no kernel image is available

今天药忘吃喽~/ 2023年09月23日 17:53/ 0 赞/ 188 阅读

相关 RuntimeError: Input, output and indices must be on the current device

我看了一下模型的输入，都放到了GPU上，然后我在网上查了一下，发现我在模型内部创建了新的模型，这些新的模型也要放在GPU上 self.bert = BertModel

小灰灰/ 2023年01月13日 04:15/ 0 赞/ 100 阅读

相关 RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 ‘target‘

一般有三点错误 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu") 1、

待我称王封你为后i/ 2022年12月08日 15:46/ 0 赞/ 346 阅读

相关 PyTorch报错：RuntimeError: CUDA error: device-side assert triggered at /pytorch/aten/src/THC/generic

训练模型报错 RuntimeError: cuda runtime error (710) : device-side assert triggered at /

小鱼儿/ 2022年11月15日 14:10/ 0 赞/ 269 阅读

相关 RuntimeError: Attempting to deserialize object on CUDA device 1 but torch.cuda.device_count() is 1.

出现这样的原因是因为我用了第二块GPU训练，在另一台机器上加载模型的时候，没有了第二块GPU，这时候就要指定用“cuda:0” torch.load('./mo

本是古典何须时尚/ 2022年08月30日 14:55/ 0 赞/ 239 阅读

相关 RuntimeError: CUDA error (10): invalid device ordinal

两张卡跑的train，在本机一张卡上test不了结果很简单，直接将 `model.module` 拿出来就行了因为之前双卡用的parallel def _

ゝ一世哀愁。/ 2022年03月09日 14:17/ 0 赞/ 422 阅读

相关 Pytorch 多GPU multi-GPU使用 RuntimeError: all tensors must be on devices[0]

问题：在跑多GPU时出现RuntimeError: all tensors must be on devices\[0\]问题，找了一番没解决，最后瞎试出来了。主体

古城微笑少年丶/ 2021年06月24日 15:59/ 0 赞/ 1890 阅读