Skip to content
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.

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

全量微调显存占用 #1

Open
ACXuLiu opened this issue Jul 5, 2023 · 8 comments
Open

全量微调显存占用 #1

ACXuLiu opened this issue Jul 5, 2023 · 8 comments

Comments

@ACXuLiu
Copy link

ACXuLiu commented Jul 5, 2023

请问大佬在全量微调时,显存占用是多少呢?

@SpongebBob
Copy link
Owner

默认的配置我用了8个40GB的A100。

@wac81
Copy link

wac81 commented Jul 7, 2023

2个40g就可以

@Alwin4Zhang
Copy link

2个40g就可以

有开源吗?

@ustbliubo2014
Copy link

2个40g就可以

这个是用的原始的配置吗?我用 4 个 A00 仍然 GPU 不够, out of memory

@resehist
Copy link

训练了多久呢?

@wac81
Copy link

wac81 commented Sep 28, 2023

用dp zero 2, 长度拉小点一般256-512就够了,我用2个40g,4个3090 都测试过,可以跑,如果跑一半不可以,就可能是长度问题。

@wac81
Copy link

wac81 commented Sep 28, 2023

2个40g就可以

这个是用的原始的配置吗?我用 4 个 A00 仍然 GPU 不够, out of memory

检查cpu的内存,不一定是显存不够

@dlxj
Copy link

dlxj commented Oct 18, 2023

用dp zero 2, 长度拉小点一般256-512就够了,我用2个40g,4个3090 都测试过,可以跑,如果跑一半不可以,就可能是长度问题。

麻烦能贴一下详细的参数配置吗,我用 4 卡 24G 的 A5000 一直爆显存, CPU 内存是 80G

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

7 participants