YoLOV3初训练遇到的一些问题

1./darknet: error while loading shared libraries: libcudart.so.10.2: cannot open shared object file: No such file or directory.

解决方法:

sudo cp /usr/local/cuda-10.2/lib64/libcudart.so.10.2 /usr/local/lib/libcudart.so.10.2 && sudo ldconfig

sudo cp /usr/local/cuda-10.2/lib64/libcublas.so.10.2 /usr/local/lib/libcublas.so.10.2 && sudo ldconfig

sudo cp /usr/local/cuda-10.2/lib64/libcurand.so.10.2 /usr/local/lib/libcurand.so.10.2 && sudo ldconfig

2.CUDA Error: out of memory     darknet: ./src/cuda.c:36: check_error: Assertion `0' failed.

内存不足问题:

解决方法:

由subdivisions=8改成subdivisions=16,64等。

cfg 文件夹下的 yolov3-voc.cfg文件部分内容,把subdivisions=8改成subdivisions==64,程序即可正常:

 

posted @ 2020-01-31 00:28  AnswerThe  阅读(1404)  评论(0编辑  收藏  举报