AI Pool
Log in

ResourceExhaustedError: OOM when allocating tensor with shape

I just downloaded a tensorflow model and started to test it on my examples, but I get this error.

ResourceExhaustedError: OOM when allocating tensor with shape

Why am I getting this, if I don't have any training?

tensorflowtensorflow-model
3 votesLP199.00
1 Answers
JO228.00
3

Your model just doesn't fit in your GPU. The memory of your GPU is not enough, that's why you're getting it.

Just try to change the batch size and make it smaller.

Also take a look at this github issue if you need more details.

Reply
Couldn't find what you were looking for?and we will find an expert to answer.
How helpful was this page?