google / in-silico-labeling

Code for In silico labeling: Predicting fluorescent labels in unlabeled images

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

std::bad_alloc in tensorflow

michalstepniewski opened this issue · comments

I am running the code on AWS r5a.xlarge machine with
  | Deep Learning AMI (Ubuntu) Version 14.0 - ami-0089d61bf6a518044-- | --

The machine has 32GB RAM and 4 vCPUs.
I run into:

2019-05-17 16:24:17.915477: W tensorflow/core/framework/allocator.cc:124] Allocation of 3328000000 exceeds 10% of system memory.

terminate called after throwing an instance of 'std::bad_alloc'
what(): std::bad_alloc

and the running seems to stall.

thanks for prompt reply. I am running on AWS on 61GB RAM and it's working so far so apparently 32GB is not enough :)