I see. I’ll have to check the code to see what values it uses in the error message.
cuDNN uses a lot less memory (and is faster). That is the reason I suggested so  Even just with CUDA it might use less memory than CPU.