win10下TensorFlow-gpu安装

  1. 安装python3.5.x版本

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

    • 先查看一下是否支持你的显卡
    • 安装时尽量默认(方便找)
    • 注意!!
      • 一定要配置环境变量CUDA_PATH和CUDA_PATH_V8_0
      • 两个值均为C:Program FilesNVIDIA GPU Computing ToolkitCUDAv8.0(此为默认安装路径,可以根据自己情况更改)
  3. 安装cudnn

    • 下载后解压缩
    • 将里面的bin、lib、include目录下的东西均对应复制到C:Program FilesNVIDIA GPU Computing ToolkitCUDAv8.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)
原文地址:https://www.cnblogs.com/renzhong/p/6925858.html