win10下TensorFlow-gpu安装

  1. 安装python3.5.x版本

    • windows下的TensorFlow仅支持Python3.5.x
    • 可以使用Anaconda

  2. 安装CUDA

    • 先查看一下是否支持你的显卡
    • 安装时尽量默认（方便找）
    • 注意！！
      • 一定要配置环境变量CUDA_PATH和CUDA_PATH_V8_0
      • 两个值均为C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0（此为默认安装路径，可以根据自己情况更改）

  3. 安装cudnn

    • 下载后解压缩
    • 将里面的bin、lib、include目录下的东西均对应复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0对应目录中

  4. 安装tensorflow-gpu

    • pip3 install tensorflow-gpu

  5. 测试TensorFlow

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

 

posted @ 2017-05-31 20:41  超任  阅读(505)  评论(1)  编辑  收藏  举报