MaskRCNN使用tf-gpu环境搭建实战

在服务器上搭建环境，而服务器上的cuda版本很迷惑，想使用tensorflow-gpu必须要使用对应版本的cuda和cudnn，看了很多教程觉得最简单的就是在虚拟环境下装cuda和cudnn再装tf-gpu。

我选用了cuda9.0+cudnn7.1.2+tensorflow-gpu1.12.0

1.创建虚拟环境

conda create -n cuda9 python=3.6
#激活虚拟环境
source activate cuda9

 2.添加清华源并使用conda安装cuda和cudnn（使用conda安装保证了cuda安装在虚拟环境下，不影响系统默认环境下的cuda）

#加入清华源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes

#安装cuda
conda install cudatoolkit=9.0 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/
 
#安装cudnn
conda install cudnn=7.1.2

到这里cuda和cudnn就安装完毕了，非常简单，可以conda list查看一下是不是安装了对应版本。 

3.安装tensorflow-gpu

建议先安装一个合适的numpy和h5py，否则import tensorflow时会有一些warning,我目前发现numpy1.16.0和h5py2.10.0是不会报错的。

pip install numpy==1.16.0
pip install h5py

conda install tensorflow-gpu

我好像是使用conda安装的，用pip也行，这个影响应该不大，安装时我没有指定版本。

conda install tensorflow-pgu后会显示将要安装的版本，如果对应的版本明显有问题就按n退出安装，然后指定1.12.0版本，如果相差不大的话可以不管。直接y安装

安装完成后可以通过import tensorflow来验证安装的tf-gpu有没有问题。

4.安装maskrcnn需要的包

pip install scipy
pip install Pillow
pip install cython
pip install matplotlib
pip install scikit-image==0.16.2
pip install keras==2.2.4
pip install opencv-python
pip install imgaug
pip install ipython

注意keras和tensorflow版本是相对应的。

我标注了版本号的包是我试错时出现过问题的，按照我标注的版本号安装一搬不会出问题。

5.克隆Mask_RCNN

cd workdir进入你想安装的目录（workdir是我的的安装目录）

git clone https://github.com/matterport/Mask_RCNN.git

6.安装pycocotools

先克隆pycocotools的文件夹

git clone https://github.com/cocodataset/cocoapi.git

克隆完毕后

cd 到cocoapi/PythonAPI文件夹中

python setup.py build_ext --inplace

python setup.py build_ext install

接下来测试pycocotools是否安装成功

进入ipython，输入from pycocotools import mask 没有错误提示就代表成功。

7.下载coco权重

链接：https://github.com/matterport/Mask_RCNN/releases

 我是在本地下载好后上传到服务器的，放在Mask_RCNN目录下。

以上就是搭建maskrcnn环境的全部内容，小白第一次写搭建记录，还望大佬们多多指教。

以上步骤参考：

MaskRCNN在linux服务器上搭建_qq_35807867的博客-CSDN博客_maskrcnn安装centos

ubuntu16.04 通过anaconda建立虚拟环境，安装cuda9.0，cudnn7.1.2_陌若安生的博客-CSDN博客