Mask_Rcnn_tf2在2080显卡上训练遇见的问题

在我们配置好2080显卡+cuda+cudnn+tensorflow-gpu时,由于这几个模块之间有一个版本对应问题,2080显卡要求cuda11.5(不知道cuda10是否可以,没试过),对应的tensorflow-gpu版本2.4.0   ,然而在训练mask_rcnn 时,代码一般都是tensorflow1.x ,与tensorflow不兼容。也就是说用显卡2080,对应cuda11.5,仅支持tensorflow2.x以上版本,如果需要使用tensorflow1.x ,我们需要使用低版本的cuda,对应的显卡就比较旧(不知道新的显卡对应旧版本的cuda是否兼容)

现在我们想用2080+cuda11.5+cudnn8.4.0去训练tensorflow1.x版本的mask-rcnn应该怎么做呢?

解决方法之一就是在github上下载Mask_Rcnn_TF2版本,将下载好的mrnn模块和tensorflow1.x模块替换,然而替换之后可能会出现一些问题。下面是我在配置过程中遇见的一些问题,仅作记录

1.加载完数据后,直接久退出程序了

Process finished with exit code -1073740791 (0xC0000409)

解决方法:可通过在pycharm设置一些参数解决,详情请点下面链接查看

http://t.csdn.cn/gkwZa

2.参数配置完成后,继续运行,会提示,找不到 zlibwapi.dll文件,如下图所示

解决方法:可通过到nvidia官网下载文件,将其放置在系统路径下,详细步骤请点击下面链接

http://t.csdn.cn/zZ34a 

成功解决 问题

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值