I'm trying to run a basic tensorflow lstm on aws. I have a basic EC2 ubuntu instance set up with tensorflow, as well as a python script I want to run. My script gets all the way to training (making model and processing data is ok) but then gives me this error: tensorflow.python.framework.errors_impl.ResourceExhaustedError: OOM when allocating tensor with shape[1024,1024] and type float on /job:localhost/replica:0/task:0/device:CPU:0 by allocator cpu
I think this might be an error with the EC2 setup, because the same code worked on my own machine(though I was using jupyter notebook). However, this is my first time using aws and therefore I am unfamiliar with it. Is this an error with my AWS setup and if so how do I fix it?