Clearing GPU Memory - PyTorch

[cedric (Cedric Chee)]

Try to restart the Jupyter kernel.

How to free-up GPU memory in pyTorch 0.2.x?

Yeah I just restart the kernel.

Or, we can free this memory without needing to restart the kernel. See the following thread for more info.

GPU memory not being freed after training is over

I was checking my GPU usage using nvidia-smi command and noticed that its memory is being used even after I finished the running all the code in lesson 1 as shown in the figure bellow [Capture] The memory is only freed once I restart the jupyter kernel. Is this the usual behavior? How can I free this memory without needing to restart the kernel? Thanks