Cuda out of memory

关于cuda显存不足

可以通过找代码查询当前GPU占用多少 将一些非必要的程序关闭。

或者通过修改宽度和高度的大小(32倍数)，这个方面可能会略微降低你识别的精准度。如果你的识别的类别不多，降低这些数据一般没有太大的影响。