Windows10下安装原生TensorFlow GPU版

  1. 下载 CUDA 8.0cuDNN v6 for CUDA 8.0

(下载cuDNN须要先注册NVIDIA开发帐户并登陆才能看到下载界面)python

CUDA 9 要TF 1.5版本才支持windows

  1. 安装CUDA
  2. 解压cuDNN到一个你喜欢的位置,复制文件夹的绝对路径并加到 PATH 环境变量去,而后把该文件夹下的bin文件夹的路径也放到PATH里去
  3. 安装Anaconda Python 3.5 version

注意windows tf 仅支持 python 3.5code

  1. 安装 tf

为了不以后使用 pip 安装 tf 报错,请先在cmd运行如下代码
pip install --ignore-installed --upgrade pip setuptools
以后安装tf
pip install tensorflow-gpu==1.1.0ip

接来下检验安装是否成功开发

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

另外,若是用conda的话必定要注意python版本,目前最高支持3.5get

conda create --name tensorflow python=3.5
activate tensorflow
pip install tensorflow-gpu

原文做者: filwalinecmd