Windows10下安装原生TensorFlow GPU版

  1. 下载 CUDA 8.01cuDNN v5.11

  2. 安装CUDA

  3. 解压cuDNN到一个你喜欢的位置,复制文件夹的绝对路径并加到 PATH 环境变量去,然后把该文件夹下的bin文件夹的路径也放到PATH里去

  4. 安装Anaconda Python 3.5 version
    注意windows tf 仅支持 python 3.5

  5. 安装 tf
    为了避免之后使用 pip 安装 tf 报错,请先在cmd运行以下代码

pip install --ignore-installed --upgrade pip setuptools
之后安装tf
pip install --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-0.12.0-cp35-cp35m-win_amd64.whl
或者
pip install tensorflow-gpu

接来下检验安装是否成功

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

原文作者: filwaline

相关推荐