请选择 进入手机版 | 继续访问电脑版
查看: 800|回复: 0

Pytorch释放显存占用

[复制链接]

665

主题

1234

帖子

6496

积分

xdtech

Rank: 5Rank: 5

积分
6496
发表于 2020-4-22 08:55:44 | 显示全部楼层 |阅读模式
pytorch不会主动释放显存。

第一步,删除变量
del var

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

注意:只有第一步,是不能释放显存的。
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

快速回复 返回顶部 返回列表