With this LSTM definition, the memory usage is now ~114MB.



 asha97:

Do CUDA version effect the execution of code?


The memory usage for the CUDA context might differ from different CUDA versions.
The model itself should not use more or less memory.