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.

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

Jump to bottom

CUDA out of memory #59

Open
unstory opened this issue Aug 19, 2020 · 2 comments
Open

CUDA out of memory #59

unstory opened this issue Aug 19, 2020 · 2 comments

Comments

@unstory
Copy link

unstory commented Aug 19, 2020

train_batch_size我设置成1了，为啥还是out of memory.
我的显卡 getforce 930MX , 2G显存
请问最低显存要求是多少？为什么会out of memory?
15万的训练数据，才5个标签，应该不会很吃显存吧
@lonePatient
Copy link
Owner

2g太小了啊 ，bert本身有点大，另外跟你的seq_length有关。换卡吧 8G我都觉得玩耍不友好了。

@unstory
Copy link
Author

unstory commented Aug 20, 2020

感谢回复。我没钱，不换显卡，搞完这个我就不弄nlp了。

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@unstory @lonePatient