安装tensorflow-gpu详细步骤

cuda+cudnn+tensorflow-gpu

一.安装cuda

1. 去官网https://developer.nvidia.com/cuda-toolkit-archive 选择要下载的cuda版本，如下：
   
   2.按照如下方式选择环境(这个好像不需要登录也可以下载)，选择base installer所在栏点击下载，得到cuda_9.0.176_384.81_linux.run文件。
   
   3.安装cuda：sudo sh cuda_9.0.176_384.81_linux.run
   4.安装过程中注意仔细看提示，有一项是提示：是否安装驱动driver，这一项选择No，其它的都选Yes/Accept或者回车就可以了

二. 安装cudnn

1.下载cudnn时要先注册cudnn账号。

2.去官网 https://developer.nvidia.com/rdp/cudnn-archive 选择要下载的cudnn版本，如下：
点击下载（需要登录上面cudnn账号之后才能下载）得到：cudnn-9.0-linux-x64-v7.3.0.29.solitairetheme8文件

3.用解压工具解压上述文件，如在Linux环境下：tar xvf cudnn-XXXX即可解压，得到解压之后的cuda目录；

4.把上述解压之后的cuda目录下的cudnn文件拷贝到cuda-9.0中：
sudo cp cuda/include/* /usr/local/cuda-9.0/include
sudo cp cuda/lib64/* /usr/local/cuda-9.0/lib64

三.安装tensorflow

1.修改.bashrc文件中的环境变量如下：
vi ~/.bashrc

export PATH=/usr/local/cuda-9.0/bin:$PATH

export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64$LD_LIBRARY_PATH

source ~/.bashrc

2.创建新的conda环境：conda creat -n tf-new python=2.7

3.激活conda环境：source activate tf-new

4.下载tensorflow安装包得到：XXXX.whl文件
附：下载链接：https://pypi.org/project/tensorflow/#files

5.安装tensorflow：pip install --ignore-installed XXXX.whl

备注：对与步骤4,5，也可以直接使用：pip install --upgrade tensorflow-gpu
若指定tensorflow版本 : pip install --upgrade tensorflow-gpu==1.0

这样就安装成功啦，很开心有木有！
检查：python
import tensorflow as tf
检查tensorflow是否是gpu版本的，可以在tensorflow官网上随便找个例子试试~~

四.安装过程中，常见问题

ImportError：libcublas.so.9.0

百度了很多方法均不行，最终我的解决方法是（3步缺一不可）：
1.在 ~/.bashrc 里面加上：export CUDA_HOME=/usr/local/cuda-9.0
2.直接将/usr/local/cuda-9.0/lib64/里面的libcublas.so.9.0.176直接重命名为libcublas.so.9.0
3.sudo ldconfig /usr/local/cuda-9.0/lib64
哈哈哈，比较粗暴的解决方法~~~~~

备注：该配置对应的是 cudnn 7.05，否则也可能会出错！