New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

一机多gpu卡启动codegeex4失败 #60

Open

xiaoyezi320 opened this issue Aug 13, 2024 · 1 comment

xiaoyezi320 commented Aug 13, 2024

服务器有3张T4卡，每张T4卡显存16G，codegeex4本地化部署启动至少需要19G左右到显存，启动时却只使用一张卡，报了oom异常，请问下如何才能在启动时指定使用多张卡启动模型？

hyh19962008 commented Sep 30, 2024

https://github.com/THUDM/ChatGLM-6B里有个utils.py，可以参考一下

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment