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;

猜你喜欢

转载自blog.csdn.net/qq_35898332/article/details/110388353