在安装之前说明下我的电脑配置
Ubuntu14.04
CUDA7.5
NVIDIA GT750Ti
主要参考链接是 https://github.com/weiliu89/caffe/tree/ssd ,以下是步骤在其基础上做的补充
1.Get the code.
git clone https://github.com/weiliu89/caffe.git
cd caffe
git checkout ssd
2.Build the code.
cp Makefile.config.example Makefile.config
make -j8
sudo vi ~/.bashrc
加入你自己的caffe下的python路径,我的是
export PYTHONPATH=$PYTHONPATH:/home/xk/caffe/python
source ~/.bashrc
make py
make test -j8
make runtest -j
Train参考https://github.com/weiliu89/caffe/tree/ssd注意例子训练是有预训练文件的
Test参考https://github.com/weiliu89/caffe/tree/ssd
3.Error Solution
1.error == cudaSuccess (10 vs. 0) invalid device ordinal ,解决办法:设置好GPU gpus = "0,1,2,3" 改为gpus = "0"02.error == cudaSuccess (2 vs. 0) out of memory,解决办法:调小batchsize大小 batch_size = 1 accum_batch_size = 1
2.If you have multiple GPUs installed in your machine, make runtest might fail. If so, try following:
export CUDA_VISIBLE_DEVICES=0;
make runtest -j8
3.Ifyou have error: "Check failed: error == cudaSuccess (10 vs. 0) invalid device ordinal",first make sure you have the specified GPUs, or try following if you have multiple GPUs:
unset CUDA_VISIBLE_DEVICES