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

modelscope-funasr报这个错怎么解决呀?

modelscope-funasr报这个错怎么解决呀?Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False. If you are running on a CPU-only machine, please use torch.load with map_location=torch.device('cpu') to map your storages to the CPU.

展开
收起
三分钟热度的鱼 2024-03-06 20:07:51 28 0
2 条回答
写回答
取消 提交回答
  • 面对过去,不要迷离;面对未来,不必彷徨;活在今天,你只要把自己完全展示给别人看。

    这个错误提示表明您正在尝试在CUDA设备上反序列化对象,但是torch.cuda.is_available()返回False,表示您的机器不支持CUDA。如果您正在运行在仅支持CPU的机器上,请使用torch.load函数并将map_location参数设置为torch.device('cpu'),以将存储映射到CPU。

    以下是示例代码:

    import torch
    
    # 加载模型时指定map_location为'cpu'
    model = torch.load('your_model_path', map_location=torch.device('cpu'))
    

    通过将map_location参数设置为torch.device('cpu'),您可以确保模型加载到CPU上进行计算,而不会引发CUDA相关的错误。

    2024-03-08 20:11:13
    赞同 展开评论 打赏
  • device="cpu" 升级最新版本:pip3 install -U funasr modelscope
    用法见文档:https://www.modelscope.cn/models/iic/speech_paraformer-large-vad-punc_asr_nat-zh-cn-16k-common-vocab8404-pytorch/summary 此回答整理自钉群“modelscope-funasr社区交流”

    2024-03-06 20:33:35
    赞同 展开评论 打赏

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

相关电子书

更多
低代码开发师(初级)实战教程 立即下载
冬季实战营第三期:MySQL数据库进阶实战 立即下载
阿里巴巴DevOps 最佳实践手册 立即下载