Ubuntu16.04 安装tensorflow+Fast-RCNN+cuda+cudnn过程

Ubuntu16.04 安装tensorflow+Fast-RCNN+cuda+cudnn过程

---

• 软件驱动安装

  显卡驱动：nvidia显卡驱动
  cuda安装包（下载.run安装包）：Cuda
  cudnn: cudnn(根据所使用的深度学习的平台需求确定cudnn 的版本，比如：可以在官网上查询ubuntu 16.04 tensorflow所需要的cudnn版本)

• Nvidia显卡驱动配置

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
sudo apt-get install nvidia-384 #我的是1050Ti显卡
#之后重启

输入

nvidia-smi   #会显示你显卡的信息

- Cuda配置

 chmod +x cuda_8.0.28_linux.run
 sudo ./cuda_8.0.28_linux.run  #说明：进入.run安装选项之后，针对【是否要设置driver的选项，要选NO】。

cuda环境配置

export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

安装cudnn
1. 到官网下载cudnn https://developer.nvidia.com/rdp/cudnn-download
下载cudnn5.1 for cuda8.0

2. d进入cudnn5.1解压之后的include目录，在命令行进行如下操作：

sudo cp cudnn.h /usr/local/cuda/include/    #复制头文件

1. 再将cd进入lib64目录下的动态文件进行复制和链接：

sudo cp lib* /usr/local/cuda/lib64/    #复制动态链接库
cd /usr/local/cuda/lib64/
sudo rm -rf libcudnn.so libcudnn.so.5    #删除原有动态文件
sudo ln -s libcudnn.so.5.1.10 libcudnn.so.5  #生成软衔接 libcudnn.so.5.1.10看自己的版本
sudo ln -s libcudnn.so.5 libcudnn.so      #生成软链接

1. 安装tensorflow（tensorflow-1.4需要cudnn6.0版本，1.5-最新版需要cudnn9.0的）

   pip install tensorflow-gpu(tensorflow 如果是cpu的，选这个)
   pip install tensorflow-gpu==1.4(选择版本)