window安装TensorFlow-GPU版本

TensorFlow 2.4需要CUDA 11.0，而不是 11.2或10.0）

CUDA11.0+cuDNN8.0.x
(文件一定是要对应的,CUDA cuDNN TensorFlow的版本一定要对应,不然用不了)
对于其他版本的安装也适用，替换到相应的版本即可。版本对应查看点这里。

1.CUDA11.0下载


打上这四个勾就足够了
安装好CUDA11.0;

2.cuDNN下载
解压,复制cuDNN8.0.x内的三个文件夹粘贴到cuda11.0的安装路径下;

即v11.0这个目录的子目录

效果:

打开cmd，将 CUDA和 cuDNN 安装目录添加到 %PATH% 环境变量中

以下命令只能用于cuda11.0的情况，其他版本的cuda需要将v11.0改成对应的cuda版本数

SET PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\bin;%PATH%
SET PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\extras\CUPTI\libx64;%PATH%
SET PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\include;%PATH%

cmd用pip下载TensorFlow-GPU

 pip install --upgrade --ignore-installed -i https://pypi.mirrors.ustc.edu.cn/simple tensorflow-gpu==2.4.0

验证安装效果：

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