Maskrcnn_benchmark最新安装教程以及错误解决⽅法(Ubuntu16.04)。。。
Maskrcnn_benchmark最新2019安装教程以及错误解决⽅法(Ubuntu16.04)
0、序⾔
最近要做⼀些⽬标检测、识别的相关⼯作,⽤到了何⼤神的maskrcnn,但是在配maskrcnn环境问题上出了很多错误,浪费了很多时间,特此做了安装踩坑总结,希望有所帮助。 其中借⽤了很多⼤佬的经验,链接已给出,感谢。
⼀、电脑配置如下
系统:Ubuntu 16.04
显卡:NVIDIA RTX 2070
cuda版本:10.0.130(deb⽂件安装)
cudnn版本:7.6.1
显卡驱动410.48(此为安装cuda10.0时其⾃带的版本)
GCC版本:5.4.0
ubuntu安装教程
⼆、安装步骤及遇到的错误
1、安装
(1)先按照安装,⼤部分步骤⽆问题。依个⼈经验在以下步骤需要注意。
(2)安装到第四步,即安装pytorch 与 cudatoolkit时容易出问题,本⼈电脑按照官⽅教程来安装不会成功,⽤以下指令可以成功:
conda install -c pytorch pytorch-nightly torchvision cudatoolkit=10.0
pip install torchvision==0.2.2
还有⼀种是根据安装pytorch,虽然本电脑未成功,但有问题可以试试
(3)安装到第六步,执⾏ python setup.py install --cuda_ext --cpp_ext 时,可能会出现错误。可以执⾏
python setup.py install
去掉后边的–cuda_ext --cpp_ext。可以成功。但是在我第⼆次重头再安装maskrcnn_benchmark时候,敲全部的指令 python setup.py install --cuda_ext --cpp_ext 也不会出现问题,原因未知。
(4)另外按照官⽅安装步骤安装完并没有安装cudnn,可以⾃⼰⼿动在该环境下安装cudnn 7.6的版本,指令如下 :
conda install cudnn=7.6
2、遇到的错误
(1)跑程序时出现:
ImportError:/home/ouc/.conda/envs/MaskRCNN/lib/python3.6/site-packages/torchvision/_C.cpython-36m-x86_64-linux-gnu.so: undefined symbol: _ZN2at 7getTypeERKNS_6TensorE
此问题是由于torchvision未安装好。有两种解决⽅式:
⼀为:先执⾏:
conda uninstall pytorch
conda uninstall pytorch-nightly
conda uninstall torch
codna uninstall torchvision
然后执⾏:
c onda install -c pytorch pytorch-nightly torchvision cudatoolkit=10.0
pip install torchvision==0.2.2
⼆为:通过以下指令:
git clone github/pytorch/vision.git
cd vision/
python setup.py install
重新安装后,记得重新执⾏maskrcnn_benchmark官⽅指南中的:
cd maskrcnn-benchmark
python setup.py build develop
这两种⽅法都可以解决此⽅式,但推荐第⼀种。通过第⼆种也可以解决此问题,但可能有其他遗留问题。
(2)若⽤⽅法⼆解决问题(1)后会出现,在vision⽬录下python->import torchvision没问题,换个⽬录就报错的问题。
解决⽅法:将vision下的torchvision和-info⽂件夹复制到/.anaconda/envs/Maskrcnn_benchmark/lib/python3.6/site-packages下⾯即可,删除原来的。
(3)还有⼀些在安装电脑配置时的⼀些问题,链接放在这⾥
CUDA的
安装完CUDA后,nvcc -V查不到信息,