如何通过Pytorch有效释放显存占用？

本文共计227个文字，预计阅读时间需要1分钟。

如果在Python中调用PyTorch且存在GPU内存占用，不会自动释放，需要手动释放。可以使用以下代码：

pythontorch.cuda.empty_cache()

官方文档说明：释放所有当前由ca持有的未占用的缓存内存。