ubuntu下tensorflow的GPU版本安装记录

原创 2018年04月17日 21:58:41

开始学习tensorflow了,记录一下
提前说一下前面我已经安装好了nvidia的驱动以及 cuda cudnn,没有安装的话需要根据文档安装 cuda文档地址 cudnn安装文档
下面的教程根据官方文档记录 文档地址,python3,
我安装的时候出现一个问题,解决方法是一定要安装cuda9.0 cuda9.1会有问题,tf暂时不支持

1. 安装libcupti-dev

libcupti-dev是cuda分析工具接口,要安装它需要先安装cuda-command-line-tools

sudo apt-get install cuda-command-line-tools

再将下面路径添加到环境变量中

vim /etc/sudo vim /etc/profile
# 将下面的添加到profile末尾
export LD_LIBRARY_PATH=${LD_LIBRARY_PATH:+${LD_LIBRARY_PATH}:}/usr/local/cuda/extras/CUPTI/lib64
# :wq 保存退出
source /etc/profile #让修改生效

安装libcupti-dev, 注意cuda Toolkit的版本要大于7.5

sudo apt-get install libcupti-dev

2. 安装tensorflow

官方提供了以下几种安装方法

  • Virtualenv虚拟运行环境
  • 原生pip安装
  • Docker安装
  • Anaconda安装
  • 源码安装

官方好像推荐虚拟运行环境。然后我就选择通过虚拟运行环境安装吧。

2.1 安装virtualenv

下载相关软件

sudo apt-get install python-pip python-dev python-virtualenv # Python 2.7
sudo apt-get install python3-pip python3-dev python-virtualenv # Python 3.n选择这个

创建运行环境
(我在这里设置的安装目录是 ~/.venv/tensorflow 是这个,如果大家安装的不一样要注意一下换个路径)

mkdir ~/.venv/tensorflow
virtualenv --system-site-packages ~/.venv/tensorflow  # 环境安装目录 python2下用这个
virtualenv --system-site-packages -p python3 ~/.venv/tensorflow #我这里环境安装目录 python3 下用这个

使虚拟运行环境生效(进入运行环境)

source ~/.venv/tensorflow/bin/activate

如果你的前缀变成了这样,那就安装好了

安装

2.2 安装tensorflow-gpu版本

很简单一句话

pip3 install --upgrade tensorflow-gpu # for Python 3.n and GPU

另外官方还提供其他版本的安装方法

pip install --upgrade tensorflow      # for Python 2.7 cup 版本
pip3 install --upgrade tensorflow     # for Python 3.n cpu版本
pip install --upgrade tensorflow-gpu  # python3.7 GPU版本

接下来就是漫长的下载了
这里写图片描述

另外google还提供了url安装方式,安装失败的同学可以试下

ip3 install --upgrade \
 https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-1.7.0-cp34-cp34m-linux_x86_64.whl

如果还有问题,google提供了问题帮助 如果不行,百度吧。

2.3 验证是否安装成功

首先确保是在tensorflow的安装环境中(有前缀),然后随便一个位置创建一个py文件,输入以下文件

# Python
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))

如果输出

Hello, TensorFlow!

就好了!下面是我的运行结果。
这里写图片描述

我遇到的几个问题

  1. 要用cuda9.0。而我刚开始装的是9.1,会有问题
  2. 9.0的最低支持显卡驱动384,注意下载的驱动要高一点
  3. 运行的时候要设置为gpu为nvidia的,而非笔记本的集显。(小米笔记本air)
版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/liuYinXinAll/article/details/79979612

VB中获取逻辑磁盘的信息

我们在编程的时候有时会需要得到系统中逻辑磁盘的一些信息,如磁盘卷标、磁盘序列号、空间大小、剩余空间等,这些信息直接使用VB提供的函数显然是无法得到的。但是,借助于VB对WINDOWS API函数的支持...
  • jadedrip
  • jadedrip
  • 2000-11-08 22:20:00
  • 1077

ubuntu下安装gpu版本tensorflow

这些天搞tensorflow,电脑辣鸡,cpu计算太慢,想换gpu,要装三个东西:n卡驱动(375)、cuda(8.0)、cudnn(6.0版本居然不行,必须是5.1)。卡在n卡驱动好久,还重装了了一...
  • fan624009652
  • fan624009652
  • 2017-06-05 18:07:25
  • 2182

ubuntu14.04系统中安装tensorflow(gpu版)cuda-8.0 + cudann(5.0 or 6.0)

系统配置:ubuntu14.04 + GTX1060 + i7(7700K) 参考地址:http://blog.csdn.net/zhouchao_fight/article/details/5151...
  • u012969412
  • u012969412
  • 2017-03-21 22:40:42
  • 8791

ubuntu下tensorflow的GPU版本安装记录

开始学习tensorflow了,记录一下 提前说一下前面我已经安装好了nvidia的驱动以及 cuda cudnn,没有安装的话需要根据文档安装 cuda文档地址 cudnn安装文档 ...
  • liuYinXinAll
  • liuYinXinAll
  • 2018-04-17 21:58:41
  • 71

ubuntu16.04下安装TensorFlow(GPU加速)----详细图文教程

写在前面一些废话接触深度学习已经有一段时间,之前一直在windows下使用Theano,但是发现Theano天书般的源码真是头大,在看到tensorflow中文教程后,发现它竟然逻辑清晰,教程丰富,实...
  • zhaoyu106
  • zhaoyu106
  • 2016-10-18 19:58:02
  • 129927

ubuntu下TensorFlow的安装

ubuntu下TensorFlow的安装机器学习是当前最热门的话题,相信只要是程序员都知道机器学习,机器学习的原理说真的非常的复杂,如果所有的原理都靠自己来实现的话,想要入门的话那基本把99%的程序员...
  • DilemmaVF
  • DilemmaVF
  • 2017-03-26 16:19:41
  • 783

Ubuntu下如何安装TensorFlow

本文目录引言 基于Anaconda的tensorflow安装 1 下载linux版本的Anaconda安装包 2 安装Anaconda 利用anaconda安装tensorflow 1 建立一个 co...
  • tina_ttl
  • tina_ttl
  • 2016-06-26 15:29:40
  • 66127

Ubuntu 16.04下安装Tensorflow(GPU)

ubuntu下Tensorflow的GPU配置
  • u010789558
  • u010789558
  • 2016-07-09 17:24:21
  • 11887

ubuntu16.04下安装CUDA,cuDNN及tensorflow-gpu版本过程

这篇文章主要依据两篇文章: 深度学习主机环境配置: Ubuntu16.04+Nvidia GTX 1080+CUDA8.0 深度学习主机环境配置: Ubuntu16.04+GeForce GTX ...
  • u014595019
  • u014595019
  • 2016-12-19 00:52:46
  • 18511

ubuntu14.04系统中安装tensorflow(gpu版)

原文:http://blog.csdn.net/u012969412/article/details/64502393 目录(?)[+] 系统配置:ubuntu14.04 + GTX1060 + ...
  • xiangyong58
  • xiangyong58
  • 2017-05-10 11:33:02
  • 554
收藏助手
不良信息举报
您举报文章:ubuntu下tensorflow的GPU版本安装记录
举报原因:
原因补充:

(最多只允许输入30个字)