首先给大家梳理一下TensorFlow1.6.0版本的安装步骤:
1、安装CUDA9.0
2、安装cudnn
3、安装TensorFlow-gpu1.6.0
Part I 安装CUDA
首先,确定你要安装的CUDA版本一定要与TensorFlow版本对应,否则后面安装TensorFlow会报错。
其他的TensorFlow版本我不清楚,但是1.5.0及1.6.0的TensorFlow与CUDA9.0是能够正确匹配的。
注意!!!一定要是CUDA9.0,多一点都不行!如果你安装了CUDA9.2,最后安装TensorFlow的时候就会报错。
CUDA9.0下载地址:https://developer.nvidia.com/cuda-90-download-archive
CUDA9.0安装步骤:按照提示安装就完事了。
Part II 安装cudnn
很多人会疑问为什么还要装这个,这个东西主要是用来给深度学习做加速的,是较新版本的TensorFlow-gou版的必备环境。
下载地址:https://developer.nvidia.com/cudnn
注意,下载这个包需要注册,按照提示注册好之后,下载就行了。
下面重点来了,如何安装cudnn。
STEP1 解压下载的cudnn文件
下载的cudnn大概长