Cuda11.7, 显卡Nvidia A10,运行GPT3 2.7B模型时报错

服务器显卡型号:Nvidia A10,Cuda版本11.7,在读取GPT3-2.7B模型时报错: Unsupported gpu architecture 'compute_86'

image.png 查阅过相关资料,无论Cuda版本还是显卡型号,都应该支持8.6的算力,请问是什么原因报错呢?

展开
收起
游客3jyk5rrnw6wte 2023-02-13 19:20:38 1588 分享 版权
1 条回答
写回答
取消 提交回答
  • 发表文章、提出问题、分享经验、结交志同道合的朋友

    楼主你好,报错信息已经提示了:架构不兼容 GPT-3 2.7B 模型所使用的CUDA架构为 compute_86,这个架构属于 Ampere 之前的GPU架构,所以在您的GPU上运行时会报错。

    可以尝试以下两种方法: 1、更新CUDA版本:您可以尝试更新CUDA版本到11.4或以上版本,以支持您的Ampere架构的显卡。 2、切换模型:您可以尝试使用支持您的GPU架构的GPT-3模型。您可以在Hugging Face等模型库中寻找支持您GPU架构的模型,或者自己在Colab等平台上进行训练。

    2023-02-16 14:57:56
    赞同 展开评论

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

收录在圈子:
+ 订阅

热门讨论

热门文章

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