一定是keras和tensorflow版本不匹配的问题

原因是,keras和tensorflow版本不匹配的问题

https://docs.floydhub.com/guides/environments/ 

最终,我安装的是tensorflow-gpu2.2.0 和 keras 2.3.1;你也可以选择其它组合

我是根据我安装的CUDA和cudnn来选择tensorflow-gpu版本的,如下是对应关系

 

Logo

汇聚全球AI编程工具,助力开发者即刻编程。

更多推荐