参考博客:

如何计算模型以及中间变量的显存占用大小:

https://oldpan.me/archives/how-to-calculate-gpu-memory

如何在Pytorch中精细化利用显存

https://oldpan.me/archives/how-to-use-memory-pytorch

再谈Pytorch中的显存利用问题(附完善显存跟踪代码)

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

转载于:https://www.cnblogs.com/jiangkejie/p/10098995.html

Logo

脑启社区是一个专注类脑智能领域的开发者社区。欢迎加入社区,共建类脑智能生态。社区为开发者提供了丰富的开源类脑工具软件、类脑算法模型及数据集、类脑知识库、类脑技术培训课程以及类脑应用案例等资源。

更多推荐