当前位置:   article > 正文

测试yolov3时报错:cuda error: out of memory darknet: ./src/cuda.c:36: check_error: Assertion `0' failed.

darknet: ./src/cuda.c:36: check_error: assertion `0' failed. aborted (core d

按照darknet官网上的教程安装这个框架,根据自身的条件(已装好nvidia 显卡驱动,cuda9.0 cudnn7.1),在makefile文件中修改GPU=1 CUDNN=1 OPENCV=1,然后重新make,再下载了yolov3.weights后,然后使用该网页上提供的测试命令做测试。

darknet: ./src/cuda.c:36: check_error: Assertion `0' failed.

结果报错:内存溢出。

CUDA Error: out of memory
darknet: ./src/cuda.c:36: check_error: Assertion `0' failed.
Aborted (core dumped)

error text

那么,修改cfg文件夹下的yolov3.cfg文件,原始的yolov3.cfg文件开头为:

  1. [net]
  2. # Testing
  3. #batch=1
  4. #subdivisions=1
  5. # Training
  6. batch=64
  7. subdivisions=16

修改为:

  1. [net]
  2. # Testing
  3. batch=1
  4. subdivisions=1
  5. # Training
  6. #batch=64
  7. #subdivisions=16

减少其batch数量能大大缓解内存紧张。

如果上面这样做还是出现内存不足的现象,那么只有两种方案了,要么用小一点的模型,要么就更改cfg目录下yolov3.cfg第8、9行height和width值了,我修改为宽高都为416,再测试就没问题了。修改后的yolo.cfg文件:

修改后

再来测试就没问题了。

ok

 

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/盐析白兔/article/detail/88896
推荐阅读
相关标签
  

闽ICP备14008679号