Tensorflow GPU版 安装教程

　　前一段时间按照官网给出的教程安装了一遍，最近重装又要重把运行环境重新安装一次，又折腾了好久，主要是Tensorflow官网打开太慢，仅此作为记录，以便以后使用。

参考资料：https://www.tensorflow.org/install/install_linux

 

  运行环境与安装包：

　　Ubuntu 16.04，Python 2.7，Virtualenv， Cuda8.0，Cudnn V5 for Cuda 8.0，NVIDIA显卡评分>3.0(GTX1060)

  第一步 安装Virtualenv、tensorflow：

1.安装 pip 与 virtualenv 

 $ sudo apt-get install python-pip python-dev python-virtualenv 

2.创建独立的python运行环境

 $ virtualenv --system-site-packages tensorflow

3.激活tensorflow运行环境

 $ source ~/tensorflow/bin/activate # bash, sh, ksh, or zsh

4. 安装tensorflow

(tensorflow)$ pip install --upgrade tensorflow-gpu

第二步：安装cuda与cudnn

下载地址：https://developer.nvidia.com/cuda-downloads

下载CUDA

安装CUDA：

1.  $ sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb
2.  $ sudo apt-get update
3.  $ sudo apt-get install cuda

设置环境变量LD_LIBRARY_PATH与CUDA_HOME：

打开配置文件

vim  ~/.bash_rc

在文件末尾添加下面两行

export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64"

export CUDA_HOME=/usr/local/cuda

安装Cudnn：

下载地址：https://developer.nvidia.com/cudnn

需要注册，并填写一堆表格。。。

解压 cudnn，并copy到/usr/local/cuda文件夹

1. $  tar xvzf cudnn-8.0-linux-x64-v5.1.tgz

2. $ sudo cp cuda/include/cudnn.h /usr/local/cuda/include

3. $ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64

4. $ sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn* #给所有用户添加读的权限

安装libcupti-dev库：

$ sudo apt-get install libcupti-dev

NVIDIA显卡驱动：

 

编辑日期：2017-06-04 10:51:47