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 113 分享 版权
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
    赞同 展开评论

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

热门讨论

热门文章

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