在ubuntu下安装Detectron

安装Detectron

1)前提条件:

a) NVIDIA GPU, Linux, Python2
b) Caffe2, various standard Python packages, and the COCO API

2)检查caffe2是否安装

#交互Success,则为正常
python -c 'from caffe2.python import core' 2>/dev/null && echo "Success" || echo "Failure"
#返回可用GPU的个数,则为正常
python -c 'from caffe2.python import workspace; print(workspace.NumCudaDevices())'

若此步骤返回Failure或GPU个数小于1,则回到博客 中分析出错原因。

3)安装COCO API

# COCOAPI=你想把COCO克隆到本地的路径/cocoapi
git clone https://github.com/cocodataset/cocoapi.git $COCOAPI
cd $COCOAPI/PythonAPI
# Install into global site-packages
make install
# Alternatively, if you do not have permissions or prefer
# not to install the COCO API into global site-packages
python setup.py install --user

COCOAPI=你想把COCO克隆到本地的路径/cocoapi,在ubuntu终端中输入并赋值。

4)安装Detectron

a) 下载一个Detectron项目:

`# DETECTRON=/path/to/clone/detectron
git clone https://github.com/facebookresearch/detectron $DETECTRON

b) 安装你所下载的detectron目录中的requirements.txt

pip install -r $DETECTRON/requirements.txt

c) make

cd $DETECTRON && make

5) 查看是否Detectron安装成功

python $DETECTRON/detectron/tests/test_spatial_narrow_as_op.py

若交互以下内容,则表示detectron安装成功。

在这里插入图片描述

  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值