翻译模型。我指定了device为GPU了。但是实际运行中还是cpu,有没有遇到的

翻译模型。我指定了device为GPU了。但是实际运行中还是cpu,有没有遇到的图片.png

这是什么原因呢

展开
收起
1026720306076186 2023-06-27 10:52:19 396 分享 版权
2 条回答
写回答
取消 提交回答
  • 北京阿里云ACE会长

    码中存在一个错误:在判断是否有 GPU 可用时,您使用了一个错误的字符串字面量 “cuda:@”,而不是正确的 “cuda:0” 或 “cuda:1” 等。这可能导致 PyTorch 无法正确地识别您的 GPU 设备,从而使用 CPU 进行推理。

    2023-07-18 16:04:08
    赞同 展开评论
  • 我也遇到了,请问解决了吗

    2023-07-18 13:33:12
    赞同 展开评论

包含命名实体识别、文本分类、分词、关系抽取、问答、推理、文本摘要、情感分析、机器翻译等多个领域

热门讨论

热门文章

还有其他疑问?
咨询AI助理