本文从参考了诸多Tensorflow GPU和MaskRCNN配置的博客与教程,最终在多台不同的电脑上实现了配置。本文目的在于用预训练权重来完成MaskRCNN的demo的环境搭建。
一.工具
1. win10平台 安装步骤本文不涉及
2. ubuntu16.04平台 安装步骤本文不涉及
(1.2可二选一)
3.python 3.6
官网:https://www.python.org 在download中可以下载。
4. anaconda 官方网站下载速度很慢,可以使用清华大学镜像和中科大镜像。
(官网)https://www.anaconda.com/download/
(清华镜像)https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
5. Cuda 9.0 Toolkit (版本要求极高,最新是Cuda 9.1是不兼容之后的tensorflow1.5的) (windows) https://developer.nvidia.com/cuda-90-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal(patch也需要下载,选择日期最新的)
(ubuntu) https://developer.nvidia.com/cuda-90-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1604&target_type=runfilelocal(patch也需要下载,选择日期最新的)
7. cuDNN v7.0.5 (Dec 5, 2017), for CUDA 9.0
(官网) https://developer.nvidia.com/rdp/form/cudnn-download-survey
需要先注册然后选择上面的版本要求下载。Cuda和Cudnn可能打开网页很慢,但是下载的时候速度还可以,有些用梯子的人可以考虑还是国内下载的好。
8.Keras&Theano(以防出现nomodule named keras的错误)
9.Opencv3.3.1(可用anaconda可以简单配置,windows和ubuntu略有不同,版本不一定非是3.3.1,只是windows的好安装)
10. Git(官方版下载速度很慢,可以考虑国内链接&