系统:Ubuntu 18.04
显卡:GTX 1080Ti
1.安装最新版Anaconda
2.创建cu100虚拟环境
conda create -n cu100 python=3.7
3.安装cudatoolkit 10.0,这一步需要设置好源,如果设置的源没有包含cudatolkit,这一步则安装不成功。
安装完cudatoolkit后,就尽量不用conda install命令,采用pip install,这使用过程中感觉conda install 有点迷。
conda install cudatoolkit=10.0
4.安装tensorflow-gpu, numpy, scikit-image, keras, 这一步要合适版本的,不然可能会报错或警报。
pip install tensorflow-gpu==1.13.1
pip install numpy==1.16.4
pip install scikit-image==0.16.2
pip install keras==2.0.4pip install cython
pip install imgaug
5.接下来安装pycocotools
6.代码的默认保存权重的位置为Mask_RCNN/logs
7.balloon.py同目录的两个ipynb也很有用。
参考文献:
1.Input image dtype is bool. Interpolation is not defined with bool data type 输入图像 dtype 为 bool 。未使用布尔数据类型定义插值