pytorch 的GPU性能优化

最近在训练一个图片分类器,GPU使用率一直没到100% ,甚至只是一些波峰尖刺。

后来排查了一些问题。这种情况 大概率就是 CPU的数据在往GPU搬运的时候出现了瓶颈,优化方法

1 预取数据 多线程

2 大规模映射顺序读

3 把数据搬运到ssd

最后用了3解决问题,当你的盘太慢的时候,最好的方法就是换到ssd上

你可能感兴趣的:(机器学习,系统运维,pytorch,性能优化,深度学习)