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

作者：小蓝xlanll | 2024-02-27 03:00:17

踩

runtimeerror: expected all tensors to be on the same device, but found at le

python报错：
RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0
解决办法如下

提示：针对神经网络（全连接网络）报错

一、解决办法第一步

命令行输入：数据.is_cuda
若输出Ture则，数据为gpu上的形式，反之为cpu

对数据进行转换：数据.to(device)

模型实例化后，对实例进行转化：net.to(device)

以上两步做完模型就能正常运行了！！！！！！！

（喜大普奔，来自于一个不会神经网络和python的人呜呜）

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/小蓝xlanll/article/detail/149499