相關(guān)博客：

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。

?

計算方式為：

?

?

?

?

?

?

?

總結(jié)

以上是生活随笔為你收集整理的模型占用GPU显存计算的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。