maxfiltenborg:

Strangely enough when I tried it today it does work with the same batch size and mixed precision training. Maybe I still had some allocated GPU memory…


Yes, it shouldn’t use more memory, so maybe some processes were still using some device memory.
Anyway, let us know, if you encounter any other issues.