windows系统下安装TensorFlow（GPU版）

 

windows系统下安装TensorFlow（GPU版）

 

1、python安装

 

anaconda是python的一个发行版，直接在官网下载安装包，官网地址https://www.anaconda.com/distribution/。然后开始默认安装，到下面我是勾选了两个选项。

第一个是加到环境变量里，在命令行使用需要conda命令，加入进去可以直接用了。

第二个是使ide识别conda环境。

 

另外、下载时如果很慢的话最好用清华源cmd执行这几句，（参考：https://mirrors.tuna.tsinghua.edu.cn/help/anaconda/）

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/

conda config --set show_channel_urls yes

 

安装完后在cmd下输入conda -V看一下版本号，

便安装成功，在安装目录下默认的base环境就是他本身的环境，然后你可以在envs目录下自己创建，conda create --name（ 自己起的环境名字 ）python=3.5(这里你可以指定python版本）就会创建一个环境。

 

2、看自己的显卡计算力是否符合要求（>3.5）

显卡计算力查看地址：https://developer.nvidia.com/cuda-gpus

3、查看要下载的TensorFlow版本所需的cuda和cudnn版本

1）TensorFlow版本环境匹配地址：https://tensorflow.google.cn/install/source_windows

Windows GPU 支持指南：https://tensorflow.google.cn/install/gpu

2）CUDA下载地址https://developer.nvidia.com/cuda-toolkit-archive

（下载CuDNN需要注册账号，注册过程比较简单）

4、CUDA安装

运行cuda_9.0.176_win10.exe。安装软件会先运行一个系统检查，如果没有软硬件不兼容的情况就能继续进行下一步。若有不兼容情况，系统检查则会报错，无法进行下一步。如果报错了，就最先考虑下是不是自己的显卡不被该版本支持，在选择安装项时一般不安装GeForce Experience，CUDA是核心组件必须勾上，剩下两个选项的当前版本如果比新版本低的话也可以勾选上。接着点击下一步就开始安装了。

（安装路径我选择了默认路径、这样起码不用手动更换环境变量。）

查看，nvcc -V，成功

5、CuDNN安装

解压压缩包cudnn-9.0-windows10-x64-v7.0.zip，得到三个文件夹

将这三个文件拷贝到CUDA9.0的安装路径根文件夹下

6.1 pip 安装tensorflow

pip install tensorflow-gpu==1.12.0（但不建议用pip，一直出错，不知什么原因。。。。）一开始觉得是和cuda版本不匹配、但是改过来后还是出错。

 

 

6.2直接用conda安装tensorflow

利用conda可以直接安装：（包括了cuda和cudnn）conda install tensorflow-gpu(是最高的版本)

conda install tensorflow-gpu=1.12.0将安装固定版本

import tensorflow as tf

hello = tf.constant('Hello,TensorFlow!')

sess = tf.Session()

print(sess.run(hello))

 

 至此、配置完成、这里有个不解的问题，就是tensorflow的版本是要和固定的CUDA版本匹配吗？为何我安的cuda9和tensorflow1.13（按理说应该是cuda10）也没有问题。。。。

参考：https://blog.csdn.net/weixin_39290638/article/details/80045236