报错：RuntimeError: Expected object of backend CUDA but got backend CPU for argument #3 ‘index‘

叁歲伎倆 2023-07-25 09:00 16阅读 0赞

# ![20191009191333910.png][][日萌社][Link 1] #

[人工智能AI：Keras PyTorch MXNet TensorFlow PaddlePaddle 深度学习实战（不定时更新）][AI_Keras PyTorch MXNet TensorFlow PaddlePaddle]

--------------------

RuntimeError: Expected object of backend CUDA but got backend CPU for argument #3 'index'
        分析：data数据或者model没有调用.to(device)
        解决：
            # 进行可用设备检测, 有GPU的话将优先使用GPU
            device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
            model = model.to(device) 或 model = model.cuda(device)
            output = model(input.to(device)) 或 output = model(input.cuda(device))
            # 选择损失函数, 这里选择预定义的交叉熵损失函数
            criterion = torch.nn.CrossEntropyLoss().to(device) 或 criterion = torch.nn.CrossEntropyLoss().cuda(device)
            loss = criterion(output, lable.to(device)) 或 loss = criterion(output, lable.cuda(device))

[20191009191333910.png]: /images/20230528/478bae823aef439c99b41a9854951c23.png
[Link 1]: http://www.rimengshe.com/
[AI_Keras PyTorch MXNet TensorFlow PaddlePaddle]: https://blog.csdn.net/zimiao552147572/article/details/88867161

报错：RuntimeError: Expected object of backend CUDA but got backend CPU for argument #3 ‘index‘

发表评论取消回复

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

相关阅读

相关 RuntimeError: Expected object of scalar type Double but got scalar type Float for argument #2 'mat2'

相关遇到：RuntimeError: Expected object of backend CUDA but got backend 错误应该如何解决

相关报错：RuntimeError: Expected object of backend CUDA but got backend CPU for argument #3 ‘index‘

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

相关 RuntimeError: expected backend CPU and dtype Float but got backend CPU and dtype Long

相关 BFF —— Backend For Frontend

相关 RuntimeError: Expected object of type torch.cuda.FloatTensor but found type torch.FloatTensor for ar

相关 RuntimeError: Expected object of type torch.cuda.LongTensor but found type torch.cuda.IntTensor

相关 Expected object of type torch.FloatTensor but found type torch.cuda.FloatTensor for argument

相关 expected type torch.cuda.DoubleTensor but got torch.cuda.FloatTensor python报错