mirror of
https://github.com/kvcache-ai/ktransformers.git
synced 2026-11-05 01:07:28 +00:00
[GH-ISSUE #45] Cannot run DeepSeek V2 Chat in server mode on 2 GPUs #24
Labels
No milestone
No project
No assignees
1 participant
Notifications
Due date
No due date set.
Dependencies
No dependencies set.
Reference
kvcache-ai/ktransformers#24
Loading…
Add table
Add a link
Reference in a new issue
No description provided.
Delete branch "%!s()"
Deleting a branch is permanent. Although the deleted branch may continue to exist for a short time before it actually gets removed, it CANNOT be undone in most cases. Continue?
Originally created by @ELigoP on GitHub (Aug 20, 2024).
Original GitHub issue: https://github.com/kvcache-ai/ktransformers/issues/45
I have 2x RTX 3090. I've built
ktransformersfrom source (needed to install some missing packages, but it worked in the end).When I run
local_chatwithin-console chat works, I get 9 tps, more than llama.cpp out-of-the-box 7.5 tps.
I didn't find a way to pass multi-gpu optimization rule to
ktransformersserver, as--optimize_config_path OPTIMIZE_CONFIG_PATHoption seems to be not about that. So I just replaced default file withmulti-gpuversion. The I runI get this error (in the end it says
Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cuda:1!).Should I also change
ktransformers/configs/config.yaml? How? Is the problem that onlycuda:0is mentioned in config? But right when the server starts it seems to read optimize rule right - it split 0-29, 30-59 layers bettween GPUs:The error:
@Azure-Tang commented on GitHub (Aug 21, 2024):
Sorry for the inconvenience orz.
You can appoint optimise yaml rule by
--optimize_config_pathnow. I have fixed by #48 .Expected all tensors to be on the same deviceOk this is a bug that server creating static cache on one device. Also fixed by #48 .