一步真实解决RuntimeError: CUDA out of memory.

你是不是在Pycharm等编译器和Colab等服务器上运行xxx.py

今天用pytorch训练神经网络时

出现下面的这个问题

一步真实解决RuntimeError: CUDA out of memory._第1张图片

那么我们跟着错误去上面的提示信息找方法

提示的是CUDA out of memory.可以知道咱们设置的内存使用超过了cuda的限制

那么我们将batch_size降低,将数值调小,然后再试一下。

一般是最初设置的一半,不行的话在降低一半。。。总有一个数值适合你~

然后运行下xxxx.py,发现问题解决~

备注一下:::

输入 nvidia-smi,会显示GPU的使用情况,以及占用GPU的应用程序

一步真实解决RuntimeError: CUDA out of memory._第2张图片

 小秘诀,浅浅记录录下吧

更多干货,可以关注公众号“蓝胖胖干货”,教你账号注册,快速上网等秘籍~~~咱们一起成长!

你可能感兴趣的:(真实解决Bug,python,pycharm,开发语言)