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

RuntimeError:CUDA error : out of memory #11

Open

Hzzhang-nlp opened this issue Jun 27, 2023 · 4 comments

Hzzhang-nlp commented Jun 27, 2023

Two 3060 graphics cards with a total memory of 24GB, why would this error still be reported?

Contributor

little51 commented Jun 27, 2023

using watch -n 1 "nvidia-smi",Check if other GPU is used

Author

Hzzhang-nlp commented Jun 27, 2023

This is when loading the model

Author

Hzzhang-nlp commented Jun 27, 2023

Is the wsl ubuntu we use closely related to this?

Contributor

little51 commented Jun 27, 2023

try reduce model_max_length to 256, and keep per_device_train_batch_size 1 and per_device_eval_batch_size 1

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