Windows下TensorFlow-gpu版本环境搭建

Windows下TensorFlow-gpu版本环境搭建

一、环境
1、Win10
2、显卡=1080Ti
3、Visual Studio 2015
二、TensorFlow环境搭建
1、Anaconda安装
（1）0Anaconda的下载地址：https://www.anaconda.com/products/individual。下载对应的安装包之后，点击程序一直下一步，注意的是在Advanced Installation Options（高级安装设置）中，不要勾选“Add Anaconda to the System PATH environment variable”(不将Anaconda加入环境变量中)，剩下的直接点击下一步就能完成。
（2）检查是否安装成功：conda --version
（3）检查目前安装了那些环境： conda info --envs
注意：建议将TensorFlow环境和pytorch环境分隔开。
2、cuda的安装
官网的下载地址为https://developer.nvidia.com/cuda-toolkit-archive
选择适合自己电脑的版本，本机显卡为1080ti，我选择的是cuda10.0 网盘链接为：链接：https://pan.baidu.com/s/1Y_x4hiYdSCh_G0cIe3Coyw
提取码：tc6o

显卡与版本对应的网址为：
https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
注意：在安装时安装选项选择自定义，去掉“Driver Components”选项，不更新显卡驱动。不然安装后显卡可能不能使用
3.cudnn的安装
（1）官网的下载地址为：https://developer.nvidia.com/rdp/cudnn-archive
选择适合自己电脑的版本，而且还要适配cuda。我选择的cudnn的版本为7.6.4
注意：在下载cudnn时需要自己注册一个英伟达的账号
（2）下载完成之后解压到自己想要保存的目录，并将该目录下的bin目录加入到环境变量中。
（3）解压完成之后的文件如下：

将这个三个文件夹中的文件，复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0中对应的文件夹中。
（4）.环境变量的配置

具体为：
1）C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin
2）C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\lib\x64
（5）测试
输入命令 nvcc -V（注意v是大写）

4、TensorFlow的安装
（1） 使用命令：pip install tensorflow-gpu==1.13.1
如果下载慢的话可以查看
https://blog.csdn.net/weixin_43218120/article/details/107628209
（2）测试是否安装成功
测试代码：

import tensorflow as tf
hello = tf.constant("Hello TensorFlow")
sess = tf.Session()
print(sess.run(hello))
a = tf.constant(10)
b = tf.constant(32)
print(sess.run(a+b))

测试运行的结果：