Tensorflow GPU 版本 Windows环境安装

整体安装参考：https://blog.****.net/u010099080/article/details/53418159

tensorflow-cuda-cudnn版本对应参考：https://blog.****.net/omodao1/article/details/83241074

 

1、Tensorflow-gpu版本安装之前，需要根据电脑显卡的型号下载相应版本的cuda和cudnn。

1. 安装cuda
2. cudnn包下载完，将三个文件夹中的.dll .h .lib文件分别拷贝到cuda的相应文件夹中，如图。
3. 然后将cuda的相关路径添加到电脑的环境变量中。

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin  

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64

2、根据cuda和cudnn的版本，选择匹配的tensorflow版本。

比如我的：cuda8.0 + cudnn v6.0 + python3.6 + tensorflow-gpu1.3.0

确定tensorflow版本后，在cmd或powershell中输入命令行：pip install tensorflow-gpu==1.3.0

 

 

3、测试代码

import tensorflow as tf

hello = tf.constant('hello,tensorf')

sess = tf.Session()

print(sess.run(hello))