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