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版本是没有的
测试结果如下
安装包如下