ubuntu16.04 tensorflow pip 方式安装
首先,需要知道 tensorflow 1.5版本以上包括 1.5版本 的GPU类型都是需要安装 cuda9.0的, tensorflow-gpu 1.4版本是可以使用cuda 8.0。
由于本人安装的是cuda8.0, 所以安装的tensorflow-gpu版本为 1.4 。
然而比较搞笑的是tensorflow-gpu 1.4 版本的需要的 cudnn是 6.0, 而不是cuda8.0最新支持的cudnn 7.0 。
cudnn 下载地址: v6.0版本的对应cuda8.0的那个
https://developer.nvidia.com/rdp/cudnn-archive
本博客是博主个人学习时的一些记录,不保证是为原创,个别文章加入了转载的源地址,还有个别文章是汇总网上多份资料所成,在这之中也必有疏漏未加标注处,如有侵权请与博主联系。
如果未特殊标注则为原创,遵循 CC 4.0 BY-SA 版权协议。
posted on 2018-07-08 11:23 Angry_Panda 阅读(317) 评论(0) 编辑 收藏 举报