跑通MaskRCNN的demo
整体步骤参考:
https://blog.csdn.net/wc781708249/article/details/79438972
遇到的一些问题
我的电脑事先安装CUDA8.0+CUDNN5.1,根据上面的教程安装tensorflow-gpu==1.5版本
跑demo的过程中报错libcublas.so.9.0: cannot open shared object file: No such file or directory,也就是说tensorflow版本不兼容,1.5需要CUDA9.0。
所以我选择给tensorflow降级,终端直接执行
pip install tensorflow-gpu==1.3.0
重新运行demo,报错为libcudnn.so.6:cannot open sharedobject file: No such file or directory
也就是说需要cudnn6.0版本,此处解决方案参考
https://blog.csdn.net/l297969586/article/details/67632608
当然,我也看到另一种解决方案为把tensorflow-gpu降到1.2版本,试过之后是行不通的,MaskRCNN需要tensorflow1.3及更高版本,所以只能选择升级CU