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

王小雷

发布于 2018-01-02 16:07:20

1.5K0

文章被收录于专栏：王小雷王小雷

摘要：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

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

如有侵权请联系 cloudcommunity@tencent.com 删除

ubuntu

tensorflow

本文分享自作者个人站点/博客前往查看

如有侵权，请联系 cloudcommunity@tencent.com 删除。

本文参与腾讯云自媒体同步曝光计划，欢迎热爱写作的你一起参与！

ubuntu

tensorflow

登录后参与评论

0 条评论

热度

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

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

1.安装NVIDIA的GPU-CUDA,cuDNN

1.1. 查看当前配置

1.2.安装CUDA

1.3.安装cuDNN

1.4.安装 libcupti-dev 包

1.5.变量设置

2.安装tensorflow-gpu

2.1.安装依赖

2.2.安装tensorflow-gpu版本

3.测试tensorflow-gpu

1.彩蛋-卸载方法

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

社区

活动

圈层

关于

腾讯云开发者

热门产品

热门推荐

更多推荐