CUDA 8.0 + cuDNN v6.0安装 (Linux)

CUDA 8.0 + cuDNN v6.0安装 (Linux)

低版本的tensorflow, 例如1.4.0调用GPU需要低版本的CUDA 8.0以及 cuDNN v6.0
各版本tensorflow和CUDA对应关系如下图：

下载

下载Linux对应的CUDA8.0，和对应CUDA8.0的cuDNN6.0：CUDA 8.0 NVIDIA官网；cuDNN 6.0 for CUDA 8.0 (需要注册) NVIDIA官网；

安装

参考官网Guide CUDA 8.0 for Linux; cuDNN 6.0 for CUDA 8.0 for Linux;
其他参考博客 CUDA8.0和9.0版本切换; cuDNN 6.0 安装;
笔者根据其他参考博客安装完成。

验证 tensorflow-gpu availabel

Pycharm直接install package的时候，注意要install tensorfolw-gpu
参考 验证此时的tensorflow的代码是否是在使用GPU;