- 安装cuda和cudnn,注意和tensorflow对应关系
选tensorflow-gpu==1.3.0 cuDNN 6 ,cuda 8
https://www.tensorflow.org/install/source#tested_build_configurations
- 安装tensorflow
进入虚拟环境sudo apt update sudo apt install python3-dev python3-pip sudo pip3 install -U virtualenv # system-wide install
删除pip install --upgrade tensorflow==1.3.0 pip install keras==2.0.8
requirement.txt
里的tensorflow和keraspip install -r requirement.txt
- 下载mask_rcnn_coco.h5
放到mask_rcnn的目录下 - 安装pycocotools
注意看一下cd PythonAPI make
Makefile
,如果是python3 将 Makefile 中的 python 改为 python3 - 安装jupyter
pip install jupyter
[安装配置] Mask RCNN配置教程
最新推荐文章于 2025-03-06 19:42:51 发布