pytorch中的python对象占用空间很大,模型可以用.cuda()
方法放在GPU上,临时的一个batch数据也在训练时放在GPU上进行计算,随着时间推移,显存占用指数倍增长,如果可以的话,把训练过程中的pytorch变量对象转换成numpy变量会大大减少显存的占用。
05-08
4936
![](https://csdnimg.cn/release/blogv2/dist/pc/img/readCountWhite.png)
03-09
1842
![](https://csdnimg.cn/release/blogv2/dist/pc/img/readCountWhite.png)
pytorch中的python对象占用空间很大,模型可以用.cuda()
方法放在GPU上,临时的一个batch数据也在训练时放在GPU上进行计算,随着时间推移,显存占用指数倍增长,如果可以的话,把训练过程中的pytorch变量对象转换成numpy变量会大大减少显存的占用。