empty_cache

paddle.device.cuda. empty_cache ( ) [源代码]

该函数用于释放显存分配器中空闲的显存,这样其他的 GPU 应用程序就可以使用释放出来的显存,并在 nvidia-smi 中可见。大多数情况下您不需要使用该函数,当您删除 GPU 上的 Tensor 时,Paddle 框架并不会将显存释放,而是将显存保留起来,以便在下一次申明显存时可以更快的完成分配。

代码示例

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

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