shaoheshaohe 发表于 2020-4-22 08:55:44

Pytorch释放显存占用

pytorch不会主动释放显存。

第一步，删除变量
del var

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

注意：只有第一步，是不能释放显存的。
页: [1]
查看完整版本: Pytorch释放显存占用