本文参考博客:
https://blog.csdn.net/weixin_39290638/article/details/80045236
https://www.jianshu.com/p/c4f0b7e74b40
- 首先百度搜索安装CUDA和cudnn,版本要正确。
- 安装Anaconda3,也是百度查询安装。
- 打开Anaconda3,创建运行环境,输入指令:
conda create -n tensorflow-gpu python=3.6
激活环境,输入命令:
activate tensorflow-gpu
- 安装tensorflow-gpu,输入命令:
pip install --ignore-installed --upgrade tensorflow-gpu
-
下载MaskRCNN代码,地址:https://github.com/matterport/Mask_RCNN;
-
下载coco数据集上预训练权重(mask_rcnn_coco.h5),地址:https://github.com/matterport/Mask_RCNN/releases;
-
下载pycocotools,地址: