🐕
📝FAISS GPU 版を最速で入れる方法
FAISS を GPU で動かせずに悩んでいましたが、pipでインストールできることがわかりました。
🔧 結論:このコマンドだけで完了
pip install faiss-gpu-cu12[fix-cuda]
- CUDA Toolkit の事前インストール不要
- CUDA ランタイム込みのビルドが入る
-
[fix-cuda]が環境依存の CUDA 参照問題を吸収してくれる
とにかく、これだけで GPU 版 FAISS が動く。
💡 補足メモ
- CUDA 11 系ではなく CUDA 12 系のランタイム同梱版
- Python だけ入っていれば OK
🧪 動作確認
import faiss
faiss.get_num_gpus()
GPU が 1 以上返ってくれば成功。
🎯 まとめ(自分向け)
- CUDA 入れなくていい
- 依存関係で悩まない
- とりあえずこのコマンド叩けば FAISS GPU が動く
Discussion