tensorflow2.X+2080显卡环境配置

anaconda+pycharm

python==3.6.13

keras==2.1.6

cuda==11.7(需要看显卡支持的最低版本,2080的是11.5)

cudnn==8.4.0(需要和cuda版本对应相匹配)

tensorflow-gpu==2.4.0(需要和cudnn+cuda相对应匹配,最低版本2.4.0)

如果需要使用tensorflow-gpu 1.x版本的话,需要降低cuda和cudnn的版本10+7.4

下载测试代码测试

如果出现测试module 'tensorflow' has no attribute 'constant'错误,可能是tensorflow-gpu版本低了

如果出现Could not load dynamic library ‘cusolver64_10.dll‘; dlerror: cusolver64_10.dll not found错误,GPU  fasle的话需要进入安装cuda的库里面,搜索cusolver64_11.dll,将其修改为cusolver64_10.dll

本次记录是为了跑Maskrcnn 模型,如果在训练maskrcnn时,出现

module 'tensorflow' has no attribute 'placeholder'错误,主要是因为这个版本的maskrcnn的代码使用的tensorflow1.X里的函数,这些函数在tensorflow2.x版本是没有的

测试结果如下

安装包如下

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值