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 条评论
登录 后参与评论

相关文章

来自专栏用户2442861的专栏

Caffe学习系列(12):训练和测试自己的图片

学习caffe的目的,不是简单的做几个练习,最终还是要用到自己的实际项目或科研中。因此,本文介绍一下,从自己的原始图片到lmdb数据,再到训练和测试模型的整个...

661
来自专栏小詹同学

人脸识别(三)——源码放送

人脸识别相关的基本原理和流程,以及各个步骤的介绍和代码前两篇都有介绍,其实可以通过前两篇自行整合出完整的人脸识别源码,并且适当修改可以实现MFC程...

4248
来自专栏ATYUN订阅号

小白也可以操作的手机TensorFlow教程:Android版和iOS版

TensorFlow通常用于训练海量数据,但新兴的智能手机市场也不可忽略。那些不能等待未来和love Machine Learning的人正在通过制造工具来突破...

3326
来自专栏weixuqin 的专栏

深度学习之 TensorFlow(三):TensorFlow 源代码解析

1032
来自专栏AI研习社

Github 项目推荐 | Nvidia 图片风格转换工具 —— FastPhotoStyle

FastPhotoStyle 是 Nvidia 的图片风格转换工具,其中包含了照片变为各种艺术风格的算法。通过给定的照片和风格,该工具可通过特定的算法何成为用户...

34511
来自专栏贾志刚-OpenCV学堂

tensorflow object detection API训练公开数据集Oxford-IIIT Pets Dataset

object detection API 安装参见官方的github: https://github.com/tensorflow/models/tree/ma...

1122
来自专栏机器之心

从零开始:深度学习软件环境安装指南

3468
来自专栏机器学习算法与Python学习

值得收臧 | 从零开始搭建带GPU加速的深度学习环境(操作系统、驱动和各种机器学习库)

关键字全网搜索最新排名 【机器学习算法】:排名第一 【机器学习】:排名第一 【Python】:排名第三 【算法】:排名第四 原文:https://medium....

3416
来自专栏weixuqin 的专栏

深度学习之 TensorFlow(三):TensorFlow 源代码解析

3786
来自专栏漫漫深度学习路

pytorch学习笔记(十):learning rate decay(学习率衰减)

pytorch learning rate decay 本文主要是介绍在pytorch中如何使用learning rate decay. 先上代码: def...

44210

扫码关注云+社区