关于Linux中引用auto_gptq提示“CUDA extension not installed”

wi162yyxq 2024-10-13 10:37:01 阅读 89

问题现象如下:

引用auto_gptq时报CUDA extension not installed的提示

解决方案(以下解决方案的前提是torch torchvision torchaudio都装好cu118版本了):

1、下载官方源码:

<code>git clone https://github.com/AutoGPTQ/AutoGPTQ.git

2、安装bitsandbytes

pip install bitsandbytes --no-cache-dir

3、从源码安装gptq

pip install . (Install from source AutoGPTQ)

 

然后就没问题了!



声明

本文内容仅代表作者观点,或转载于其他网站,本站不以此文作为商业用途
如有涉及侵权,请联系本站进行删除
转载本站原创文章,请注明来源及作者。