成功解决RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be
发布时间：2023-12-18 19:30
![\"\"](\"https://img.it610.com/image/info8/2d3b8229b4f64ec0b8cd1fe15381b007.jpg\")
错误代码
问题原因：这是由于Input type 与weight type的类型不符合，可以注意到Input type是没有cuda的，而weight type是在cuda上也就是有gpu的加载的。（错误代码后的or可以不用管）
解决办法：既然是类型不符导致的错误，那么就将类型转为一致，代码详见下面：
1.
device = torch.device(\'cuda:0\' if torch.cuda.is_available() else \'cpu\')
input=input.to(device)
2.
input=input.cuda()#这两种代码任选其一即可相关推荐
SpringCloud Alibaba study-4-Ribbon和LoadBalancer (微服务负载均衡器)
解决WARNING: You are using pip version 21.3.1； however, version 22.1.2 is available.You should consid
宇宙最強的IDE - Visual Studio 25岁生日快乐
python语言基础（十一）面向对象编程（类、对象、self参数、init方法、str方法、私有属性、私有方法）
rust 02 rustdesk基于rust的开源远程控制软件
【机器学习】Reinforcement Learning-强化学习学习笔记