模型占用GPU显存计算
生活随笔
收集整理的這篇文章主要介紹了
模型占用GPU显存计算
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
相關博客:
https://blog.csdn.net/wz22881916/article/details/81054036
https://blog.csdn.net/sweetseven_/article/details/96026706
https://www.cnblogs.com/vincent1997/p/10896299.html
?
詳細的GPU顯存占用可以看上面幾篇博客,為了方便記憶抽取了其中一段介紹:
正常來講,顯存的占用主要是Model本身和每一層的output。
?
計算方式為:
?
?
?
?
?
?
?
總結
以上是生活随笔為你收集整理的模型占用GPU显存计算的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 网络模型计算量评估
- 下一篇: 优图yolo-v2 loss解析(ten