0
点赞
收藏
分享

微信扫一扫

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​​


ubuntu16.04 tensorflow   pip 方式安装_Python

举报

相关推荐

0 条评论