mask-RCNN笔记——代码GPU使用

在这篇的开头,代码已经可以正常使用了,过程参考另一篇

https://blog.csdn.net/Diana_Z/article/details/83147636

 

GPU使用实验:

(这是一个使用实验,以及改错的过程)

环境:ubuntu16.04,python3.5,GPU4个

测试修改代码:train_shaoes.py

1、修改config文件

出现错误

定位之后发现错误位置在创建模型这一步

在判断GPU的时候,包导入时正确的

进入ParallelModel的构造函数,执行self.gpu_count=gpu_count的时候一切正常,并且成功生成了属性gpu_count

但是执行完成这一步之后,数据当中自动生成了一个叫做inner_model的变量,并且在将keras_model赋值给self.inner_model的时候无法成功赋值

解决方法:

加入了红框所示的一句话

2、安装TensorFlow-gpu

出现错误:libcublas.so.9.0:cannot open shared object file: No such file or directory

原因:

tensorlow的版本与cuda不一致导致的错误,查看文献后发现我使用的cuda8.0,cuDNN6.0,对应的tensorflow版本是1.3.0或者1.4.0

3、出现问题: softmax() got an unexpected keyword argument 'axis'

【1】

4、问题:reduce_mean() got an unexpected keyword argument 'keepdims'

解决:【2】

系统函数中reduce_mean()中的参数名称是keep_dims

只要把model.py文件中的出错的keepdims修改成系统函数中的名字keep_dims就可以了。

参考资料:

【1】https://blog.csdn.net/w5688414/article/details/80145894

【2】https://blog.csdn.net/qq_36362060/article/details/80888949

发布了75 篇原创文章 · 获赞 21 · 访问量 7万+
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 大白 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览