Pytorch释放显存占用

665 主题	1234 帖子	6670 积分

Rank: 5 Rank: 5

发表于 2020-4-22 08:55:44 | 显示全部楼层 |阅读模式

pytorch不会主动释放显存。

第一步，删除变量
del var

第二步，释放显存
torch.cuda.empty_cache()

注意：只有第一步，是不能释放显存的。