ragflow icon indicating copy to clipboard operation
ragflow copied to clipboard

[Question]: Cannot utilize GPU even with docker-compose-gpu.yml

Open Lycnkd opened this issue 1 year ago • 2 comments

Describe your problem

In the parsing files and reranking phase of the chat, the GPU cannot be utilized (nvidia-smi shows 0% utilization), meaning that all models built into the Docker cannot use the GPU even after using the GPU version of the yml. I'd like to ask how this issue can be resolved? Or does ragflow have strict requirements regarding the CUDA version?

Lycnkd avatar Aug 09 '24 08:08 Lycnkd

Utilization of GPU memory is 0, isn't it? Are you using BCE/BGE embedding? Otherwise, GPU will not be used.

KevinHuSh avatar Aug 09 '24 10:08 KevinHuSh

屏幕截图 2024-08-09 185039 I am using the BGE embedding, and I deploy a llama locally and use xinferece to incorporate it into RAGflow. However, except for llama inference, no other tasks will utilize the GPU.

Lycnkd avatar Aug 09 '24 10:08 Lycnkd