abetlen / abetlen/llama-cpp-python

Even after installing CUDA + Pytorch + LLamaCPP Python I see BLAS = 0 always

オープン
#1,315 コメント 2 件 リアクション 0 件 担当者 0 名 GitHub で見る
主要言語
Python
スター
10.6k
フォーク
1.4k
PR マージ指標
PR 指標を取得中

説明

I want to run my gguf model to use the GPU for inference, So for this I have done following things:

1. Installed Visual Studio Community Version 2022
2. Installed Visual Studio Build Tools
3. Installed CUDA Toolkit 12.1
4. Installed CuDNN 12X
5. Created New Conda Environment
6. Installed Pytorch compatible with CUDA
7. Checked if import torch >> torch.cuda.is_available() >> True
8. Check if CUDA Path is available in system variables. >> Available
9. I installed llama cpp python with following commands:
set CMAKE_ARGS=-DLLAMA_CUBLAS=on
set FORCE_CMAKE=1
pip install llama-cpp-python --force-reinstall --upgrade --no-cache-dir --verbose
10. There is no error in Installation either.
11. But When I run the model, it return BLAS = 0 in the console.

Whatever inferences it makes, it does on the CPU instead of GPU.

I want my model to do the inferences with GPU instead of CPU.
I have 2 Nvidia RTX A5000 GPUs.

So Kindly guide why it is happening?

PS: I tried with Reinstalling CUDA Toolkit.

AVX = 1 | AVX_VNNI = 0 | AVX2 = 1 | AVX512 = 0 | AVX512_VBMI = 0 | AVX512_VNNI = 0 | FMA = 1 | NEON = 0 | ARM_FMA = 0 | F16C = 1 | FP16_VA = 0 | WASM_SIMD = 0 | BLAS = 0 | SSE3 = 1 | SSSE3 = 0 | VSX = 0 | MATMUL_INT8 = 0 |

コントリビューションガイド

コントリビューションガイドを開く

評価

この issue はまだ評価されていません。

新しい issue をメールで受け取る

初心者向けの GitHub issue を短くまとめたダイジェスト。