Mask RCNN笔记：环境安装篇(Win10+tensrflow-gpu) 不需要另行安装cuda cudnn

Mask RCNN系列文章目录

Mask RCNN笔记：环境安装篇(Win10+tensrflow-gpu) 不需要另行安装cuda cudnn
MaskRCNN可视化界面开发(PyQt5)

---

---

前言

毕设是做关于深度学习的，之前一直在用tensorflow cpu版本，（cpu版是为了调试代码，跑代码我是通过google colab来完成的）最近室友的2070S主机正好闲着了，开始学着装gpu版本。

上网看了一下，大多数tensorflow-gpu的安装分三部分：Anaconda的安装、CUDA+cudnn的安装、tensorflow-gpu的安装。这些安装之间的版本要保持匹配，挺晕的。

但其实，利用Anaconda，是可以一步到位的。
文本默认各位读者已经安装好了Anaconda，并下载好MaskRCNN源码。
Anaconda下载链接：https://www.anaconda.com/download/
MaskRCNN源码链接：https://github.com/matterport/Mask_RCNN
本机配置：Win10+gtx2070S

---

第一步：创建虚拟环境并安装所需包

本人已经测试过在Win10+python3.6+tensorflow——gpu1.15+cuda10的配置，可以完美支持。

1.打开Anaconda Prompt

2.创建一个虚拟环境：

conda create -n maskRCNN python=3.6

完成之后会出现下面的界面

3.激活刚刚建好的虚拟环境 maskRCNN

conda activate maskRCNN

4.先配置清华镜像！

不然下载过程不仅很慢，还容易断开连接！配置清华镜像，需要以下代码

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 --set show_channel_urls yes

5.安装tensorflow_gpu 1.15版本

conda install tensorflow-gpu==1.15

到这我们可以看到程序自动为我们安装了相应版本的cuda+cudnn
输入y后开始，中途千万别关！

6.测试

安装完成后启动python环境测试

import tensorflow as tf
tf.__version__

出现上图说明安装成功

7.安装maskRCNN其他所需的包和库

cmd终端运行
先cd到maskRCNN源码里的requirements.txt,对requirements.txt进行一些修改，去掉tensorflow>=1.3.0,keras与scipy指定版本，修改后requirements.txt如下图

然后

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple -r requirements.txt

第二步：pycocotools 安装

1.获取源码

默认已经安装了git,没有git下载zip也行，就是下载慢且不稳定。

git clone https://github.com/pdollar/coco.git

2.编译

下载好后进行解压，且cd到解压后的PythonAPI文件夹目录下

# install pycocotools locally
python setup.py build_ext --inplace
 
# install pycocotools to the Python site-packages
python setup.py build_ext install

经过上面三个步骤，已经安装好了pycocotools到python库了，进入python3命令行就可以直接导入了，执行下面命令不报错，类似下图说明已经安装成功了。

总结

今天就先写到这里了，有空继续补充！！！
要早点睡！！！
如果阅读本文对你有用，欢迎点赞评论收藏呀！！！