软件清单:
安装环境:Windows10_64位
Python:3.6
TensorFlow:1.8.0
CUDA:CUDA9.0
cuDNN:cuDNN7.0
安装过程
1.安装python
建议安装Anaconda科学计算平台来管理你的python环境,里面自带了很多模板,下载很简单,直接下一步就好,这里就不赘述。
(若之前有其它的python版本,最好卸载了,以免不必要的麻烦)
官方下载:https://www.anaconda.com/download/
2.安装TensorFlow
安装好上述的python3.6环境后,在cmd中
输入:pip install tensorflow-gpu==1.8.0
这里默认的是下载最新版本的,本次安装是2018年7月8日下载,版本号是1.8.0。(注意,tensorflow有多个版本,可能会有多种安装方式,一定要找到对应的cuda和cuDNN版本下载,可以参照这里的安装版本)
3.安装cuda
安装cuda的时候一定要安装与当前tensorflow版本相匹配的cuda版本,这里安装的cuda版本应该是9.0的。
官方地址:https://developer.nvidia.com/cuda-toolkit-archive
4.安装cuDNN7
安装cuDNN库文件,这里可能需要填一个小问卷,随便填就好,下载的是一个压缩包。(下载之前需要注册账号,注册过程相当简单)
下载地址:https://developer.nvidia.com/rdp/cudnn-download
5.关联CUDA与cuDNN
解压cuDNN压缩包后,把解压后的文件分别复制粘贴到cuda对应的文件夹里。
6.测试
直接用cmd进入python命令行界面
输入:import tensorflow
sess = tensorflow.Session()
就可以看到如下信息。