Hi, i see similar issue during use CUDA.
And I try to check GPU memory, it will goes up until ~229MB. Python then can work normally.
I am not sure why. If you find a way to fix it, please tell me. thanks