你好,游客 登录 注册 搜索
背景:
阅读新闻

Ubuntu 16.04 安装 TensorFlow(GPU支持)

[日期:2017-04-25] 来源:Linux社区  作者:Linux [字体: ]

本文记录Ubuntu 16.04安装Tensorflow步骤,也包括怎么从源码编译安装Tensorflow。

要想安装Tensorflow GPU版本,你需要有一个新一点的Nvidia显卡。

Tensorflow CPU版本的安装

使用虚拟环境(可选):Python虚拟环境(pyvenv、virtualenv)

 

安装Tensorflow:

如果要升级Tensorflow,替换新版本的TF_BINARY_URL。https://www.tensorflow.org

Ubuntu 16.04 安装Tensorflow(GPU支持)

编译安装Tensorflow(GPU支持)

安装NVidia显卡驱动,你可以在Ubuntu内置的附加驱动中安装。

安装CUDA:

安装CudNN V5:https://developer.nvidia.com/cudnn

在~/.bashrc文件中添加环境变量:

使环境变量生效:

下载tensorflow源码:

安装一些编译和依赖工具:

安装Bazel:

配置编译选项:

需要输入Python路径,默认是 /usr/bin/python。如果你使用Python3,输入:/usr/bin/python3.5。

输入Python模块路径,默认是/usr/local/lib/python2.7/dist-packages。如果你使用Python3,输入:/usr/local/lib/python3.5/dist-packages。

输入Cuda SDK版本和Cudnn版本:8.0、5.1.5。

配置完成,输入如下信息:

编译tensorflow:

构建pip包:

安装pip包:

参考:

本文永久更新链接地址http://www.linuxidc.com/Linux/2017-04/143115.htm

linux
本文评论   查看全部评论 (0)
表情: 表情 姓名: 字数

       

评论声明
  • 尊重网上道德,遵守中华人民共和国的各项有关法律法规
  • 承担一切因您的行为而直接或间接导致的民事或刑事法律责任
  • 本站管理人员有权保留或删除其管辖留言中的任意内容
  • 本站有权在网站内转载或引用您的评论
  • 参与本评论即表明您已经阅读并接受上述条款