PyTorch节省显存占用方法

1-使用inplace操作

2-使用混合精度运算

参考:

[1]混合精度训练

http://kevinlt.top/2018/09/14/mixed_precision_training/

[2]pytorch节省显存小技巧

pytorch节省显存小技巧_向阳争渡的博客-CSDN博客_pytorch减少显存

[3]Pytorch有什么节省内存(显存)的小技巧

Pytorch有什么节省显存的小技巧? - 知乎

posted @ 2022-11-13 22:36  dlhl  阅读(112)  评论(0)    收藏  举报