Ubuntu16.04配置Mask R-CNN环境(不安装Anaconda)

概述:

由于在Ubuntu16.04中ROS和Ananconda冲突,不能同时安装,所以这里尝试直接在Ubuntu16.04上配置Mask RCNN环境。

升级Python版本到3.7

Ubuntu16.04自带的是Python3.5.2,新版本的numpy等一些库要求python版本要大于3.6,所以这里将python3.5.2升级到python3.6

# 官网下载安装包
1. wget http://www.python.org/ftp/python/3.6.4/Python-3.6.4.tgz

# 解压 
2. tar -xvzf Python-3.6.4.tgz

# 安装
3. cd Python-3.6.4
4. ./configure --with-ssl

# 编译
5. sudo make
6. sudo make install

# 创建软连接,先查看python3.6.4的安装路径
7. which python3    # /usr/local/bin/python3
8. sudo  ln -s /usr/local/bin/python3 python

到此安装成功,依次输入python2,python3.5,python3.6可以查看ubuntu上的三个python版本,输入python3可以看到默认的python3为3.6

安装显卡驱动+Cuda+Cudnn+Tensorflow

这里请参考我的另一篇博客Ubuntu16.04配置Mask-Rcnn环境
操作一模一样,照着做就好。

安装MaskRCNN

同样按照上面的博客安装,下面记录一下安装过程中的错误及其解决办法。

  • 安装pycocotools
git clone https://github.com/waleedka/coco.git
cd /home/shiqi/coco/PythonAPI
python3 setup.py build_ext install

执行上述命令时会发生**“ImportError: No module named Cython.Build”**问题,但是在之前的步骤中已经通过pip3 install cpython安装过了,这里出现错误的原因是系统中存在了Python3.5和Python3.6两个版本,而上述命令默认将其安装到了python3.5目录下了。输入下面命令即可解决:

sudo python3.6 -m pip install cpython

运行MaskRCNN

  • 运行官方实例
cd /home/shiqi/Mask_RCNN
jupyter notebook

选中demo.ipynb文件,运行即可。

  • 运行自己的程序

讲mask_rcnn_coco.h5拷贝到MaskRCNN文件夹下,将demoshiqi.py拷贝到/MaskRCNN/samples文件夹下,运行命令:

python3 demoshiqi.py

此时可能会报错:Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR
解决办法:在要执行的python文件中添加下列语句,这里是demoshiqi.py

from tensorflow.compat.v1 import ConfigProto
from tensorflow.compat.v1 import InteractiveSession

config = ConfigProto()
config.gpu_options.allow_growth = True
session = InteractiveSession(config=config)

你可能感兴趣的:(Ubuntu环境搭建,python,ubuntu,自动驾驶,tensorflow,深度学习)