CPU环境下导入报错

已解决

在CPU环境下导入,报以下错误:/home/anaconda3/envs/modelscope/lib/python3.7/site-packages/modelscope/models/nlp/gpt3/distributed_gpt3.py in 21 from megatron.model import (AttnMaskType, Float16Module, LayerNorm, 22 bias_gelu_impl) ---> 23 from megatron.model.fused_softmax import FusedScaleMaskSoftmax 24 from torch import nn 25 from torch.nn import functional as F 116 def _get_cuda_bare_metal_version(cuda_dir): --> 117 raw_output = subprocess.check_output([cuda_dir + '/bin/nvcc', '-V'], 118 universal_newlines=True) 119 output = raw_output.split() TypeError: unsupported operand type(s) for +: 'NoneType' and 'str' 请问要怎么处理

展开
收起
wallowlip 2023-02-24 10:12:42 801 分享 版权
1 条回答
写回答
取消 提交回答
  • 推荐回答

    您好 ,这个模型规模比较大,不支持CPU环境load呢,可以在GPU试下

    2023-02-24 10:34:57
    赞同 展开评论

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

热门讨论

热门文章

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