Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same

作者：我家自动化 | 2024-02-18 01:36:09

踩

input type (torch.floattensor) and weight type (torch.cuda.floattensor) shou

在运行torch中出现这个错误。

错误内容大概就是指输入类型是CPU（torch.FloatTensor），而参数类型是GPU（torch.cuda.FloatTensor）。
关于数据类型的链接：官方链接

首先，请先检查是否正确使用了CUDA。

通常我们这样指定使用CUDA：


device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
inputs.to(device)

这样就把input这个tensor转换成了CUDA 类型。
但是我们还是出错。就是输出本篇博文的标题。

正确的做法是：


device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
inputs = inputs.to(device)

原因：
tensor.to() 这个函数功能是产生一个新的tensor，并不会改变原数据。

但是，注意到
Module.to() 是一个“in-place”方法，tensor.to() 函数不是。

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