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 #6

Open
lamiaa-menshawy opened this issue Aug 15, 2024 · 0 comments
Open

CUDA out of memory #6

lamiaa-menshawy opened this issue Aug 15, 2024 · 0 comments

Comments

@lamiaa-menshawy
Copy link

When I run the DKNet and DKNET-cl
Training finished but at test phase the next message appear
OutOfMemoryError: CUDA out of memory. Tried to allocate 9.00 GiB. GPU 0 has a total capacity of 4.00 GiB of which 0 bytes is free.
Size of image 128*128
Batch size 32
Learning rate 0.0001
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
1 participant
@lamiaa-menshawy