Closed1

auto_gptqのcuda extention not installed警告を回避した時の個人用メモ

悪虫悪虫

雑にpip install auto-gptqしたらcuda extention not installedとか言われた。
これを回避したいならwhlからinstallしよう。

 $ pip install auto_gptq-0.2.2+cu118-cp310-cp310-win_amd64.whl

ダウンロードリンク
https://github.com/PanQiWei/AutoGPTQ/releases

poetry環境なら以下

pyproject.toml
[tool.poetry.dependencies]
python = "^3.10"
auto-gptq = {url = "https://github.com/PanQiWei/AutoGPTQ/releases/download/v0.2.2/auto_gptq-0.2.2+cu118-cp310-cp310-win_amd64.whl"}
torch = {url = "https://download.pytorch.org/whl/cu118/torch-2.0.0%2Bcu118-cp310-cp310-win_amd64.whl"}
transformers = "^4.30.2"
このスクラップは2023/07/13にクローズされました