AI Pool
Log in
Discussions>ResourceExhaustedError: OOM when allocating tensor with shape>

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?

Deep LearningComputer VisionAI Development/Engineering
tensorflowtensorflow-model
3 votesLP262.00
1 Answers
JO295.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?