Running out of memory

[robjlyons]

Hi, I am running this on a personal computer with a 3070ti and 32gb of ram.

I am immediately running out of memory with a error saying it can't allocate 9gb of memory.

What are the system requirements for this or do I need to optimise my setup? I am using the packages and versions in requirements.txt.

Thanks.

[felix-kerkhoff]

Hi,

thank you for pointing out that issue. I am very sorry that you are having problems running my code.

Your system should be perfectly capable of running the code.

However, there might be a problem with tensorflow allocating all the gpu memory by default. Maybe limiting the gpu memory growth could be a solution (see https://www.tensorflow.org/guide/gpu#limiting_gpu_memory_growth).

Also make sure that your installed versions of tensorflow, cuDNN and CUDA are compatible (see https://www.tensorflow.org/install/source#gpu).
I specified tensorflow-gpu==2.1.0 in the requirements.txt file as I was using an older GPU with CUDA version 10.X. I guess your GPU is using CUDA 11.X, so you might have to install a newer tensorflow version like 2.5 or higher.

I hope that my answer could help you running the code on your system.

Best regards,
Felix