Ubuntu17.04安装TensorFlow1.2的GPU版本

摘要:Ubuntu17.04安装TensorFlow1.2的GPU版本。首先验证nvidia显卡,然后安装CUDA Toolkit 8.0,安装cuDNN v5深度神经网络计算加速库,最后通过python pip安装TensorFlow-GPU版本。

1.安装NVIDIA的GPU-CUDA,cuDNN

1.1. 查看当前配置

xiaolei@wang:~$ uname -m && cat /etc/*release

1.2.安装CUDA

【CUDA官方下载】https://developer.nvidia.com/cuda-downloads

xiaolei@wang:~/Downloads$ sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb
xiaolei@wang:~/Downloads$ sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-cublas-performance-update_8.0.61-1_amd64.deb
xiaolei@wang:~/Downloads$ sudo apt update
xiaolei@wang:~/Downloads$ sudo apt install cuda

1.3.安装cuDNN

NVIDIA cuDNN是用于深度神经网络的GPU加速库。

【原文】The NVIDIA CUDA® Deep Neural Network library (cuDNN) is a GPU-accelerated library of primitives for deep neural networks. cuDNN provides highly tuned implementations for standard routines such as forward and backward convolution, pooling, normalization, and activation layers. cuDNN is part of the NVIDIA Deep Learning SDK.

【官方下载】(需要注册登录后填写个小问卷才能下载)https://developer.nvidia.com/cudnn

  • tensorflow-gpu r1.2现在(201707)默认的是cuDNN v5,而v6会报错。
  • 安装方式很简单,下载后解压,把cudnn中的文件内容拷贝到对应的cuda中。
xiaolei@wang:~/Downloads$ tar -zxf cudnn-8.0-linux-x64-v5.1.tgz && cd cuda
xiaolei@wang:~/Downloads/cuda$ sudo cp include/cudnn.h /usr/local/cuda-8.0/include/
xiaolei@wang:~/Downloads/cuda/lib64$ sudo cp lib64/libcudnn* /usr/local/cuda-8.0/lib64

1.4.安装 libcupti-dev 包

xiaolei@wang:~$ sudo apt-get install libcupti-dev

1.5.变量设置

#!/bin/sh
# Author:wangxiaolei 王小雷
# Blog: http://blog.csdn.net/dream_an
# Github: https://github.com/wangxiaoleiai
# Date: 201707
# Organization: https://github.com/whaleai

export CUDA_HOME=/usr/local/cuda-8.0
export PATH=$CUDA_HOME/bin:$PATH
export LD_LIBRARY_PATH=$CUDA_HOME/lib64

2.安装tensorflow-gpu

可以通过python2或者python3的安装,博主使用的是python3。

2.1.安装依赖

xiaolei@wang:~$ sudo apt-get install python-pip python-dev   # for Python 2.7
xiaolei@wang:~$ sudo apt-get install python3-pip python3-dev # for Python 3.n

2.2.安装tensorflow-gpu版本

xiaolei@wang:~$ pip install tensorflow-gpu  # Python 2.7;  GPU support
xiaolei@wang:~$ spip3 install tensorflow-gpu # Python 3.n; GPU support

3.测试tensorflow-gpu

xiaolei@wang:~$ python3
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))

可以看到已经使用GPU了!完结-成功

1.彩蛋-卸载方法

xiaolei@wang:~$ sudo pip uninstall tensorflow # for Python 2.7 xiaolei@wang:~$ sudo pip3 uninstall tensorflow # for Python 3.n

2.彩蛋-官网给出的常见错误处理方法

https://www.tensorflow.org/install/install_sources#common_installation_problems

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏AI研习社

将 TensorFlow 训练好的模型迁移到 Android APP上(TensorFlowLite)

最近在做一个数字手势识别的APP(关于这个项目,我会再写一篇博客仔细介绍,博客地址:一步步做一个数字手势识别APP,源代码已经开源在github上,地址:Chi...

2253
来自专栏专知

浅显易懂的分布式TensorFlow入门教程

【导读】分布式TensorFlow可以有效地提神经网络训练速度,但它的使用并不简单。虽然官方提供了文档和示例,如链接【1】,但是它们太难懂了。本文是一篇浅显易懂...

2157
来自专栏简书专栏

深度学习环境搭建-CUDA9.0、cudnn7.3、tensorflow_gpu1.10的下载

本文作者接触深度学习2个月后,开始进行目标检测实践。 本文作者的专题《目标检测》链接:https://www.jianshu.com/c/fd1d6f784c...

1242
来自专栏磐创AI技术团队的专栏

Tensorflow多GPU使用详解

1.4K4
来自专栏深度学习思考者

Python常用第三方模块PIL(anaconda下载)

前提:针对深度学习的用户,推荐下载anaconda,其下载地址如下: https://repo.continuum.io/archive/.winzip/ ...

2659
来自专栏深度学习与计算机视觉

Windows系统配置TensorFlow教程

我为什么选择Windows系统: 其实我们没有必要过多纠结与到底使用Linux系统还是Windows系统来学习TensorFlow,因为不管是那个系统我们关...

3476
来自专栏梦里茶室

TensorFlow深度学习笔记 Tensorboard入门

Github工程地址:https://github.com/ahangchen/GDLnotes 官方教程:https://www.tensorflow.org...

2008
来自专栏AI研习社

Github 项目推荐 | 用于可视化人工神经网络的 Python 库 —— ANN Visualizer

ANN Visualizer 是一个很不错的 Python 库,兼容 Keras,它使用 Python 的 graphviz 库来创建开发人员正在构建的神经网络...

983
来自专栏简书专栏

Windows系统查看CUDA版本号

在按住Win键的情况下,按P键,呼唤出搜索框。 在搜索框中输入control panel,如下图所示:

2442
来自专栏Gaussic

Caffe for Windows 配置与测试

转载请注明出处:http://my.oschina.net/gaussik/blog/600368

752

扫码关注云+社区