2021-01-19

首先需满足计算机显卡为NVIDIA，否则不能安装
NVIDIA显卡的计算力大于3.5，计算力查询网址https://developer.nvidia.com/cuda-GPUS
NVIDIA驱动程序版本需418.x或更高版本，可在cmd命令窗口中nvidia-smi查询

1.下载CUDA

版本CUDA Toolkit 10.1
网址：https://developer.nvidia.com/cuda-toolkit-archive
下载后 安装按默认即可

2.安装cudnn

https://developer.nvidia.com/
版本cudnn-10.1-windows10-x64-v8.0.5.39.zip
然后将zip中的四个文件都复制到CUDA的安装目录下

配置环境变量

非常重要！！！
点击“此电脑”，右键选择“属性”，在“控制面板主页”中选择“高级系统设置”，在弹出的对话框中点击“环境变量”，进入环境变量编辑对话框。

在用户变量对话框，选中PATH，点击编辑，浏览路径
将CUDA和cudnn的路径加入
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\lib\x64

3.安装tensorflow

首先安装好Anaconda，去官网下载即可打开anaconda prompt

conda create -n tensorflow2 python==3.7
conda activate tensorflow2
pip install tensorflow-gpu==2.1 

打开Anconda Navigator,切换通道，但后点击Spyder的install。需要用到什么就安装哪个

4.测试是否安装成功

import tensorflow as tf
tf.compat.v1.disable_eager_execution()
hello=tf.constant('hello tensorflow!')
sess = tf.compat.v1.Session()
print(sess.run(hello))

输出

'hello tensorflow!'

安装成功。