服务器显卡型号:Nvidia A10,Cuda版本11.7,在读取GPT3-2.7B模型时报错: Unsupported gpu architecture 'compute_86'
查阅过相关资料,无论Cuda版本还是显卡型号,都应该支持8.6的算力,请问是什么原因报错呢?
楼主你好,报错信息已经提示了:架构不兼容 GPT-3 2.7B 模型所使用的CUDA架构为 compute_86,这个架构属于 Ampere 之前的GPU架构,所以在您的GPU上运行时会报错。
可以尝试以下两种方法: 1、更新CUDA版本:您可以尝试更新CUDA版本到11.4或以上版本,以支持您的Ampere架构的显卡。 2、切换模型:您可以尝试使用支持您的GPU架构的GPT-3模型。您可以在Hugging Face等模型库中寻找支持您GPU架构的模型,或者自己在Colab等平台上进行训练。