跑通MaskRCNN的demo

跑通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

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值