操作方法
首先是安装python,我这里安装的是python3.6版本
记得添加python的环境变量
激活python3.6的环境:activate python3.6
在cmd中输入pip install tensorflow-gpu
接着安装cuda8.0,在网上下载安装包即可。目前tensorflow只支持cuda8.0,其他版本的都安装不成功。
安装完之后验证一下cuda8.0是否安装成功
接着安装cuDNN5.1forcuda 8.0版本,然后解压生成三个文件夹
分然后将cuda/include、cuda/lib、cuda/bin三个目录复制到到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0对应的include、lib、bin目录下
在系统环境变量CUDA_PATH中添加如下变量: C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin; C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64
测试tensorflow-gpu是否安装成功; import tensorflow as tf hello = tf.constant('Hello, TensorFlow!') sess = tf.Session()