Caffe_SSD训练错误集锦

最新推荐文章于 2020-12-09 22:54:14 发布

置顶清风与归_G

最新推荐文章于 2020-12-09 22:54:14 发布

阅读量392

点赞数

分类专栏：深度学习

本文链接：https://blog.youkuaiyun.com/qq_30999155/article/details/88181928

版权

深度学习专栏收录该内容

3 篇文章

订阅专栏

本文汇总了在使用Caffe进行SSD训练时遇到的两个常见错误：1) 'No module named _caffe'，解决方案是添加环境变量并重新编译pycaffe；2) 'Check failed: error == cudaSuccess (10 vs. 0) invalid device ordinal'，修正方法是调整solver.prototxt和ssd_pascal.py中的GPU设置。参考提供的链接了解更多详情。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1. No module named _caffe

import caffe
Traceback (most recent call last):
File “”, line 1, in
File “caffe/init.py”, line 1, in
from .pycaffe import Net, SGDSolver, NesterovSolver, AdaGradSolver, RMSPropSolver, AdaDeltaSolver, AdamSolver
File “caffe/pycaffe.py”, line 13, in
from ._caffe import Net, SGDSolver, NesterovSolver, AdaGradSolver,
ImportError: No module named _caffe

解决方法：

为caffe添加环境变量，并重新编译pycaffe

sudo gedit ~/.bashrc 
export PYTHONPATH=~/caffe/python:$PYTHONPATH
source ~/.bashrc
sudo make pycaffe

2. Check failed: error == cudaSuccess (10 vs. 0) invalid device ordinal

这个问题的原因是设置的训练GPU数量和真实物理机上的显卡数量不相等，回到solover文件下，或者caffe的python执行文件，设置运行的GPU数量和指定GPU即可。

将solver.prototxt文件中的device_id项改为自己的GPU块数，0/1/2/3

device_id=0

将ssd_pascal.py文件中第332行gpus = "0，1，2，3"的GPU选择改为gpus = "0"

gpus = "0"

由于博主只有一块GPU且电脑运行内存有限，为了避免出现“Check failed: error == cudaSuccess (2 vs. 0) invalid …”的错误。还需要将ssd_pascal.py文件中的337行batch_size = 32和338行accum_batch_size = 32都改小一倍，即更改批量大小，