软硬件要求（点击前往官网查看）

安装流程（必须按照此顺序安装）

显卡驱动

1. 下载对应的驱动

2. 安装

   
   
   1. 选择NVIDIA图形驱动
   2. 精简安装即可

VS 2015-2019 Runtime

1. 安装
2. 重启

CUDA

1. 下载对应版本
2. 自定义安装（仅选择CUDA）

3. nvcc -V检测一下是否安装成功

cuDNN

1. 下载对应版本
2. 解压出来（有3个文件夹）
3. 拷贝到对应位置即可（将解压的3个文件夹拷贝到CUDA的路径下就好了）

Copy \cuda\bin\cudnn*.dll to C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\vx.x\bin.
Copy \cuda\include\cudnn*.h to C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\vx.x\include.
Copy \cuda\lib\x64\cudnn*.lib to C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\vx.x\lib\x64.

TensorFlow2.4.1 安装

pip install --upgrade tensorflow

测试

python -c "import tensorflow as tf;print(tf.reduce_sum(tf.random.normal([1000, 1000])))"

正常运行：

参考

1. CUDA官方文档：https://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/index.html
2. cuDNN官方文档：https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html#install-windows