你是不是在Pycharm等编译器和Colab等服务器上运行xxx.py
今天用pytorch训练神经网络时
出现下面的这个问题
那么我们跟着错误去上面的提示信息找方法
提示的是CUDA out of memory.可以知道咱们设置的内存使用超过了cuda的限制
那么我们将batch_size降低，将数值调小，然后再试一下。
一般是最初设置的一半，不行的话在降低一半。。。总有一个数值适合你~
然后运行下xxxx.py，发现问题解决~
备注一下：：：
扫描二维码关注公众号，回复：
14454967 查看本文章
输入 nvidia-smi，会显示GPU的使用情况，以及占用GPU的应用程序
小秘诀，浅浅记录录下吧
更多干货，可以关注公众号“蓝胖胖干货”，教你账号注册，快速上网等秘籍~~~咱们一起成长！