问题
出错代码：
DEVICE = torch.device(‘gpu’ if torch.cuda.is_available() else ‘cpu’)
报错信息：
RuntimeError: Expected one of cpu, cuda, xpu, mkldnn, opengl, opencl,
ideep, hip, ve, ort, mlc, xla, lazy, vulkan, meta, hpu device type at
start of device string: gpu
原理
无法识别“gpu”这个字符串，只能识别“cpu, cuda, xpu, mkldnn, opengl, opencl, ideep, hip, ve, ort, mlc, xla, lazy, vulkan, meta, hpu”这样的字符串
解决
将“gpu”改为训练用的设备字符串，我用的是cuda，所以写为
device = torch.device(‘cuda’ if torch.cuda.is_available() else ‘cpu’)
参考文章：https://blog.csdn.net/u014264373/article/details/87640753
衍生问题
出错代码：
if torch.cuda.is_available():#是否可以gpu加速
correct += (predict.gpu() == labels.gpu()).sum()
else:
correct += (predict == labels).sum()
报错信息：
AttributeError: ‘Tensor’ object has no attribute ‘gpu’
原理
device选择的加速设备名称是Tensor的合法属性。由于我使用的设备是cuda，则gpu应该改为cuda。
解决
将“gpu”改为训练用的设备字符串。根据使用的设备名确定修改的属性名称。文章来源：https://www.toymoban.com/news/detail-641068.html
if torch.cuda.is_available():#是否可以gpu加速
correct += (predict.cuda() == labels.cuda()).sum()
else:
correct += (predict == labels).sum()文章来源地址https://www.toymoban.com/news/detail-641068.html
到了这里，关于Expected one of xxx device type 解决方法的文章就介绍完了。如果您还想了解更多内容，请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章，希望大家以后多多支持TOY模板网！
![【小程序】警告: [Component] property xxx of xxx received type-uncompatible value解决办法](https://imgs.yssmx.com/Uploads/2024/02/518016-1.png)
