安装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