首先,需要知道 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的那个
首先,需要知道 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的那个