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
ダウンロードリンク
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にクローズされました