ssd 安装

最新推荐文章于 2024-04-05 08:30:24 发布

算法学习者

最新推荐文章于 2024-04-05 08:30:24 发布

阅读量6.8k

点赞数

分类专栏： DL

本文链接：https://blog.csdn.net/AMDS123/article/details/53258667

版权

DL 专栏收录该内容

158 篇文章 3 订阅

订阅专栏

详见 https://github.com/weiliu89/caffe/tree/ssd

Get the code. We will call the directory that you cloned Caffe into $CAFFE_ROOT
```
git clone https://github.com/weiliu89/caffe.git
cd caffe
git checkout ssd
```

Build the code. Please follow Caffe instruction to install all necessary packages and build it.

# Modify Makefile.config according to your Caffe installation.
cp Makefile.config.example Makefile.config
make -j8
# Make sure to include $CAFFE_ROOT/python to your PYTHONPATH.
make py
make test -j8
make runtest -j8
# 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
# If you 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