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

Resource exhausted #39

Open
muluogithub opened this issue Nov 2, 2020 · 1 comment
Open

Resource exhausted #39

muluogithub opened this issue Nov 2, 2020 · 1 comment

Comments

@muluogithub
Copy link

NVIDIA GEFORCE RTX 2060 6GB+cuda10.0,运行过程中会出现资源耗尽的错误,请问是什么原因呢?

tensorflow.python.framework.errors_impl.ResourceExhaustedError: OOM when allocating tensor with shape[8,64,256,256] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc
[[{{node mish_4/mul}}]]

@Mahmood-Hussain
Copy link

The reason can be your input image size, batch size, and GPU memory. If you try to reduce image size and batch size you will not encounter this error.

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

2 participants