empty_cache¶

paddle.device.cuda. empty_cache ( ) [source]

Releases idle cached memory held by the allocator so that those can be used in other GPU application and visible in nvidia-smi. In most cases you don’t need to use this function, Paddle does not release the memory back to the OS when you remove Tensors on the GPU, Because it keeps gpu memory in a pool so that next allocations can be done much faster.

Examples

>>> 
>>> import paddle
>>> paddle.device.set_device('gpu')

>>> tensor = paddle.randn([512, 512, 512], "float")
>>> del tensor
>>> paddle.device.cuda.empty_cache()