pytorch中torch.cuda.empty

it2023-02-21  79

Pytorch已经可以自动回收我们不用的显存,类似于python的引用机制,当某一内存内的数据不再有任何变量引用时,这部分的内存便会被释放。 但有一点需要注意,当我们有一部分显存不再使用的时候,这部分释放的显存通过Nvidia-smi命令是看不到的

我的理解:释放显存

参考: https://oldpan.me/archives/pytorch-gpu-memory-usage-track

https://blog.csdn.net/qq_29007291/article/details/90451890

https://blog.csdn.net/zxyhhjs2017/article/details/92795831

最新回复(0)