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

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

日萌社" class="reference-link">20191009191333910.png日萌社

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

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

发表评论

表情：
评论列表 （有 0 条评论，129人围观）

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

相关阅读