开发者社区 > ModelScope模型即服务 > 自然语言处理 > 正文

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

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

这是什么原因呢

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

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

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

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

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

相关电子书

更多
阿里巴巴高性能GPU架构与应用 立即下载
GPU在超大规模深度学习中的发展和应用 立即下载
RDS SQL Server CPU高使用率性能优化 立即下载