码迷,mamicode.com
首页 > 系统相关 > 详细

ubuntu16.04 tensorflow pip 方式安装

时间:2018-07-08 11:49:50      阅读:216      评论:0      收藏:0      [点我收藏+]

标签:下载地址   方式   tensor   1.4   idt   evel   版本   需要   方式安装   

首先,需要知道   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 方式安装

标签:下载地址   方式   tensor   1.4   idt   evel   版本   需要   方式安装   

原文地址:https://www.cnblogs.com/devilmaycry812839668/p/9279514.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!