如何在ubuntu下安装detectron2_ubuntu16.04安装detectron教程
ubuntu16.04安装detectron教程
系统环境要求:NVIDIA GPU, Linux, Python2
Caffe2, 部分Python包, COCO API
1. python2.7
Detectron必须在python2环境,python3不支持, 推荐使用conda 创建一个新的环境python27,并且切换到新环境
conda create -n python27 python=2.7
conda activate python27
以下所有操作都在python2.7下执行
2. 安装 CUDA 8 + CuDNN 7 + NCCL
2.1 CUDA8安装
安装过程略
验证安装:nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2016 NVIDIA Corporation
Built on Tue_Jan_10_13:22:03_CST_2017
Cuda compilation tools, release 8.0, V8.0.61
2.2 从Debian文件安装CuDNN 7
# 1.Install the runtime library, for example:
sudo dpkg -i libcudnn7_7.0.3.11-1+cuda9.0_amd64.deb
# 2.Install the developer library, for example:
sudo dpkg -i libcudnn7-devel_7.0.3.11-1+cuda9.0_amd64.deb
# 3.Install the code samples and the cuDNN Library User Guide, for example:
sudo dpkg -i libcudnn7-doc_7.0.3.11-1+cuda9.0_amd64.deb
2.3 测试CuDNN 7
样例在/usr/src/cudnn_samples_v7路径下
1.Copy the cuDNN sample to a writable path.
$cp -r /usr/src/cudnn_samples_v7/ $HOME
2.Go to the writable path.
$ cd $HOME/cudnn_samples_v7/mnistCUDNN
3.Compile the mnistCUDNN sample.
$make clean && make
4.Run the mnistCUDNN sample.
$ ./mnistCUDNN
显示如下,表示安装成功: Test passed!
2.4 cuDNN从v6升级到v7
cuDNN v7可以与之前版本的cuDNN共存,例如v5或v6。 cuDNN v7 can coexist with previous versions of cuDNN, such as v5 or v6.
2.5 NCCL安装
下载安装包并安装
sudo dpkg -i nccl-repo-ubuntu1604-2.2.13-ga-cuda8.0_1-1_amd64.deb
3. 安装Caffe2
3.1 conda安装Caffe2
使用Anaconda在GPU + CUDA8 + CuDNN7环境下
conda install pytorch-nightly cuda80 -c pytorch
3.2 验证Caffe2
检查Caffe2的GPU依赖是否正确,下面命令输出的GPU卡的数量必须要大于0 ,否则不能使用Detectron
# To check if Caffe2 build was successful
python -c 'from caffe2.python import core' 2>/dev/null && echo "Success" || echo "Failure"
# To check if Caffe2 GPU build was successful
# This must print a number > 0 in order to use Detectron
python -c 'from caffe2.python import workspace; print(workspace.NumCudaDevices())'
验证过程显示Failure,terminal重新输入 python -c 'from caffe2.python import core',可以显示错误信息
错误1: No module named google.protobuf.internal 解决方法:pip install protobuf
错误2: no moudle named past.builtins 解决方法:pip install future
4. 安装COCO API
# COCOAPI=/path/to/clone/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
在make install 如出现错误error: pycocotools/_mask.c: No such file or directory:
解决方法:pip install cython
5. 安装Detectron
5.1 安装
Clone the Detectron repository:
# DETECTRON=/path/to/clone/detectron
git clone https://github.com/facebookresearch/detectron $DETECTRON
Install Python dependencies:
pip install -r $DETECTRON/requirements.txt
Set up Python modules:
cd $DETECTRON && make
Check that Detectron tests pass (e.g. for SpatialNarrowAsOp test):
python $DETECTRON/detectron/tests/test_spatial_narrow_as_op.py
5.2 运行Detectron
可以使用tools目录下内置的infer_simple.py 来使用预训练的模型来预测实际的照片,infer_simple.py里面调用的是detectron封装的vis_utils.vis_one_image API。
python tools/infer_simple.py \
--cfg configs/12_2017_baselines/e2e_mask_rcnn_R-101-FPN_2x.yaml \
--output-dir /tmp/detectron-visualizations \
--image-ext jpg \
--wts https://dl.fbaipublicfiles.com/detectron/35861858/12_2017_baselines/e2e_mask_rcnn_R-101-FPN_2x.yaml.02_32_51.SgT4y1cO/output/train/coco_2014_train:coco_2014_valminusminival/generalized_rcnn/model_final.pkl \
demo
最终,检测结果就以pdf的格式输出到了/tmp/detectron-visualizations目录下
如何在ubuntu下安装detectron2_ubuntu16.04安装detectron教程相关推荐
- Win10 安装 Ubuntu16.04 双系统及 Ubuntu 下常用软件的安装(2022.3更新)
Win10 安装 Ubuntu 16.04 双系统 下载 Ubuntu 镜像文件:官网,选择 ubuntu-16.04.5-desktop-amd64.iso (1)下载 UltraISO软碟通,对一 ...
- ubuntu安装python库_Python安装:如何在ubuntu下安装python库
上面介绍到了python安装中如何在Windows下按章python库,接下来上海python培训小编介绍另外一种方法:如何在ubuntu下安装python库. 在ubuntu下进行python库的安 ...
- Ubuntu下如何使用虚拟机安装WindowsXP?(2)【转】
Ubuntu下如何使用虚拟机安装WindowsXP?(2) windowsXP系统下使用虚拟机安装ubuntu系统可能对于大多数用户来说不是问题,但是ubuntu系统下如何使用虚拟机安装windows ...
- 【Ubuntu系统配置】006Ubuntu20.04安装QQ for Linux v3.0(del文件安装卸载)
[Ubuntu系统配置]006Ubuntu20.04安装QQ for Linux v3.0(del文件安装卸载) 文章目录 [Ubuntu系统配置]006Ubuntu20.04安装QQ for Lin ...
- Win10下将Ubuntu16.04安装在移动固态硬盘上的若干问题
Win10下将Ubuntu16.04安装在移动固态硬盘上的若干问题 一.下载Ubuntu16.04系统镜像: 二.制作U盘启动盘 : 三.移动硬盘分区 : 四.安装Ubuntu系统: 1. 启动选项 ...
- ubuntu下tesseract 4.0安装及参数使用
tesseract是一个开源的OCR引擎,最初是由惠普公司开发用来作为其平板扫描仪的OCR引擎,2005年惠普将其开源出来,之后google接手负责维护.目前稳定的版本是3.0.4.0版本加入了基 ...
- ubuntu docker一键安装mysql_mysql5.6在ubuntu下的docker中安装的办法详细说明
软件安装:装机软件必备包 SQL是Structured Query Language(结构化查询语言)的缩写.SQL是专为数据库而建立的操作命令集,是一种功能齐全的数据库语言.在使用它时,只需要发出& ...
- libevent 源码阅读 Ubuntu下muduo库的安装与使用
libevent源码分析--代码结构_鱼思故渊的专栏-CSDN博客 LINUX实战:Ubuntu下muduo库的安装与使用 LINUX实战:Ubuntu下muduo库的安装与使用-LINUX入门-维易 ...
- Ubuntu下selenium+Chrome的安装使用
Ubuntu下selenium+Chrome的安装使用 安装 chrome 官网下载安装包 sudo dpkg -i google-chrome-stable_current_amd64.deb wh ...
最新文章
- 在虚拟机上安装Docker并运行Nginx
- RE:大家说说开发的时候类名和文件名一般是怎么规范的?
- hdu 527 Necklace
- Xcode中release和debug模式
- Hadoop自学笔记(三)MapReduce简单介绍
- 赚钱就是一个合作的过程,合作的过程,也是扯淡的过程
- webstorm如何自定义代码模板
- 谁设计python_如果你有设计师朋友,请对TA好一些...
- 如何用RIGOL任意函数发生器产生任意波形
- python第一行输入整数n、然后输入n行 每行三个字符串_B站2019秋招编程题思路解析[题目要素过多]...
- 第四讲 Linux常用命令
- 高举5G和AI两面旗帜:紫光展锐市场峰会火爆申城
- zimbra更换服务器域名
- 使用sql服务器发送贺卡_发送免费电子贺卡的最佳网站
- 书单|普通员工到职场精英,这一步怎样跨越?
- CodeForces 711B. Chris and Magic Square(水题)
- 各台平台电商API接口吊桶,API接口分类
- 时光机特效在哪里?推荐这三个实用软件给你
- 解决vscode电脑卡顿问题
- 农业物联网发展有哪些问题