-
Notifications
You must be signed in to change notification settings - Fork 412
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
运行run_classifier的时候oom了 #43
Comments
和这个对显存的要求是一致的:
|
同样2080GPU 8G显存 |
后来早点原因了吗。或能被其他应用占了显存? |
已经找到原因了,当一次OOM运行过后,需要手动释放内存。如果不kill掉进程,参数设置的再小还依旧是OOM。 |
@zhengchang231 您的问题解决了吗,我也遇到了这个问题,对于bert_large模型batch_size=2,max_len=32,都OOM,显存11G. |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
请问模型对最低显存有要求吗?seq_len和batch_size等我都调到1了还是oom,我的显存是8G的,跑的是large版的,谢谢
The text was updated successfully, but these errors were encountered: