@albanD Hello, I ran into issue CUDA out of memory.
RuntimeError: CUDA out of memory. Tried to allocate 960.00 MiB (GPU 0; 15.78 GiB total capacity; 13.96 GiB already allocated; 747.44 MiB free; 13.98 GiB reserved in total by PyTorch)
13.98GB is reserved by PyTorch, so the model is unable to run. I’ve read to restart session to free up memory. What’s the accepted way to ‘restart session’? I am running my model in a docker with pre-assigned resources.