ollama icon indicating copy to clipboard operation
ollama copied to clipboard

The usage of VRAM has significantly increased

Open lingyezhixing opened this issue 1 year ago • 5 comments

What is the issue?

In previous versions, I set the context length of each of my models to the maximum value that could be fully loaded onto the GPU memory. However, after the update, I found that parts of them were being partially loaded onto the CPU instead. I wonder what could be causing this. The following table is some examples.

NAME SIZE PROCESSOR
glm4:9b-chat-2K-q5_K_M 8.3 GB 10%/90% CPU/GPU
glm4:9b-chat-10K-q4_K_M 7.8 GB 7%/93% CPU/GPU
codegeex4:9b-all-10K-q4_K_M 7.8 GB 7%/93% CPU/GPU
qwen2:7b-instruct-19K-q5_K_M 8.3 GB 13%/87% CPU/GPU
internlm2:7b-chat-v2.5-8K-q5_K_M 7.7 GB 4%/96% CPU/GPU
llama3:8b-instruct-5K-q6_K 8.2 GB 10%/90% CPU/GPU
My graphics card is a 4060 laptop model, with only 8GB of VRAM. Interestingly, even before the update, none of the models was actually utilizing the full capacity of my GPU memory.

OS

Windows

GPU

Nvidia

CPU

AMD

Ollama version

0.2.3

lingyezhixing avatar Jul 13 '24 11:07 lingyezhixing

Same situation here! This issue makes me have to stay on 0.2.1.

xuexiaojingquan avatar Jul 15 '24 08:07 xuexiaojingquan

Same situation here! This issue makes me have to stay on 0.2.1.

I am also stuck in version 0.2.1

lingyezhixing avatar Jul 16 '24 07:07 lingyezhixing

Same here. WSL2 + NVIDIA GPU

sbera77 avatar Jul 18 '24 07:07 sbera77

Is there any update? With 0.3.0 I am still on:

offloading 79 repeating layers to GPU
llm_load_tensors: offloaded 79/81 layers to GPU

for qwen2:

llm_load_print_meta: model type       = 70B
llm_load_print_meta: model ftype      = Q3_K - Large
llm_load_print_meta: model params     = 72.71 B
llm_load_print_meta: model size       = 36.79 GiB (4.35 BPW)
llm_load_print_meta: general.name     = Qwen2-72B-Instruct

for 0.2.1 I could load all in vram.

chrisoutwright avatar Jul 25 '24 16:07 chrisoutwright

I added "num_gpu":81 to the params file of the model and now it loads all of it!

chrisoutwright avatar Jul 25 '24 16:07 chrisoutwright