菜单

Ubuntu 14.04 安装配置 TensorFlow

2017年4月5日 - Deep Learning, Tools    2,035 阅读    暂无评论

  最近想玩TensorFlow,于是冒着笔记本重装系统的风险把TensorFlow装了进来。为什么说是冒险呢,因为最新版本的tf需求cuda 8.0,每次装这些跟显卡相关的东西我就头疼,一个不小心显卡驱动又跪了什么都得重装。不过凭借多年作死经验,这次很顺利就把最新版本的tf装了。

升级cuda 7.0到cuda 8.0,安装cudnn 5.1

  首先卸载cuda 7.0:

  在System Settings->Software & Updates->Additional Drivers把显卡的最新的闭源驱动装了,重启之后正常的话,显卡驱动就是装好了。cuda 8.0下载之后是一个.run文件,给它用chmod +x添加可执行属性之后用sudo运行,别安装里面的驱动,只装cuda tookits。
  cudnn在这里下载,要注册一个账号,看清楚了Tensorflow目前最新版本是1.1.0-rc0,需求cuda 8.0和cudnn 5.1 (5.0不行)。cudnn 5.1下载以后解压可以看到cuda文件夹,按照下面命令把相应头文件和库文件拷贝到系统路径下面就行了。

  TensorFlow的英文主页教程里面安装的版本落后于跟github工程里面的最新版本,其实我就是按照教程里面的pip安装就直接可以用了,不过为了找最新版本的tf源,又折腾了一下。
  看主页的GET STARTED,把必备的组件都装了,然后它推荐用pip安装是这样子的命令:

  其实想装的tf路径在TF_BINARY_URL这里标注了,看后面的字符串可以猜测这是对应的版本,直接打开这个地址,可以看到有好多安装源,我找到了笔记本ubuntu的TensorFlow 1.1版本的信息:
linux/gpu/tensorflow_gpu-1.1.0rc0-cp27-none-linux_x86_64.whl
  显然linux/gpu表示linux平台的gpu版本,1.1.0rc0表示版本号,后面的none我猜是通用平台的意思,因为我试过其他字串的,pip都表示不行。然后用pip安装就行了:

  补充,后来发现官网有更新的安装说明,说明了版本号的问题,看这里

2,035 阅读    暂无评论

发表评论

邮箱地址不会被公开。 必填项已用*标注