Win10 TensorFlow-GPU-1.9.0+CUDA9.0+cuDNN7.1 配置

安装TensorFlow-GPU首先要弄清楚其和CUDA以及cuDNN的版本对应关系。

Python版本:   Python 3.6

选择的对应关系:   TensorFlow-GPU-1.9.0+CUDA9.0+cuDNN7.1

首先安装CUDA9.0 下载地址: CUDA

将安装目录下的   \bin  和  \lib\x86 路径添加到环境变量 CUDA_PATH 中

然后安装cuDNN7.1(需要英伟达账号) 下载地址: cuDNN

解压到CUDA的安装路径

最后安装TensorFlow-GPU-1.9.0  CMD下输入:

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple --upgrade tensorflow-gpu

为什么选择TensorFlow-GPU

您必须从以下 TensorFlow 类型中选择其一来进行安装：

• 仅支持 CPU 的 TensorFlow。如果您的系统没有 NVIDIA® GPU，则必须安装此版本。请注意，此版本的 TensorFlow 通常更容易安装（用时通常在 5 或 10 分钟内），所以即使您拥有 NVIDIA GPU，我们也建议先安装此版本。预编译的二进制文件将使用 AVX 指令。
• 支持 GPU 的 TensorFlow。TensorFlow 程序在 GPU 上的运行速度通常要比在 CPU 上快得多。因此，如果您的系统配有满足以下所示先决条件的 NVIDIA® GPU，并且您需要运行性能至关重要的应用，则最终应安装此版本。

TensorFlow中文社区：Tensorflow

补充：由于现在最新的Tensorflow-GPU版本为1.10.0 所以直接 --upgrade 安装的就不是1.9.0了 ，会导致运行错误。

           可以从PyPi上下载 Tensorflow-GPU-1.9.0离线包手动安装  Release history          下载地址：1.9.0