Tensorflow学习——Win10安装Tensorflow-gpu和CUDA及cuDNN

环境：Window10

Tensorflow安装官方链接：https://tensorflow.google.cn/install/install_windows

cuda安装官方链接：https://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/

一、前期准备：

1. 安装python3，下载网站：https://www.python.org/downloads/
2. 安装numpy
3. 检查电脑GPU是否达到要求，否则安装CPU版本，检查网站：https://developer.nvidia.com/cuda-gpus
4. 安装Visual C++，若安装了Visual Studio则可不管这项

二、安装CUDA和cuDNN

    因Tensorflow-gpu对CUDA和cuDNN版本有所要求，根据安装的Tensorflow-gpu版本下载对应CUDA和cuDNN版本，可参照下图。

cuda下载链接：https://developer.nvidia.com/cuda-downloads

cudnn下载链接：https://developer.nvidia.com/rdp/form/cudnn-download-survey（需要注册填调查问卷）

下载后默认安装即可，安装完成后将目录\CUDA\v9.0\bin添加到系统路径Path中。

三、安装Tensorflow-gpu

    本文采用pip的方式安装，打开Windows Powershell（建议打开管理员模式），运行下面命令，默认安装最新版本Tensorflow-gpu，若速度慢可以改用镜像。

pip3 install --upgrade tensorflow-gpu

四、测试

    打开Windows Powershell，输入下面命令。

python

>>> import tensorflow as tf
>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess = tf.Session()
>>> print(sess.run(hello))

若获得以下结果则为安装成功。

Hello, TensorFlow!

备注：

常见问题解决方法：https://www.tensorflow.org/install/install_windows#common_installation_problems