如何在windows10上安装tensorflow-gpu

在windows上安装tensorflow-gpu，安装不上，一直提示
setuptools.extern.packaging.requirements.InvalidRequirement: Expected end or semicolon (after name and no valid version specifier)
python_version>"3.7",
只安装了tensorflow=2.1.3,这里是成功的
但tf默认是没开启GPU的,可以按win+r进入cmd,启动你的虚拟环境，输入python用以下命令检测：

'''
import tensorflow as tf
tf.test.is_gpu_available()
'''

如果最后显示false,就是没打开。
如果打开了，就是tensorflow可以使用GPU
好，按顺序说下如何在windows下安装tensorflow-gpu

硬件环境准备。

1,划重点:只能用cuda tool kit11.1
在查了官方说明后，在windows10/11上要安装tensorflow-gpu只能用cuda tool kit11.1,更高版本不再支持，如果你一定要装更高版本的只能在wsl2安装。
cuda tool kit11.1下载
下载，解包安装，点击按按默认安装。
下载时需要注册nvidia帐号，注册很麻烦，但安全性很高。
安装完成应该能看见环境变量时增加如图所示，

nvcc_test.png

(查找环境变量方法，win+i,进入系统设置，到底部系统信息，高级系统设置,环境变量)，

安装成功后，win+R进入cmd,输入nvidia-smi应该看到下述版本号：12.2

nvidia-smi.png

我的版本是12.2就是因为前面先装了12.2，后来卸裁了，重装了11.1版本，但这里还是显示12.2,这里重点是：如果输入nvidia-smi看得到cuda took kit版本号就可以认为成功了。如果没看到重启下试试。

2,安装cudnn,按照官方说明，cuda tool kit 11.1对应的版本cudnn版本为8.0,在下面这个链接
cudnn安装

找到你的版本对应的cudnn版本，我的是8.0.5，下载后是个zip文件，解压zip文件，得到三个目录，把这三个目录拷贝到：
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1
这里是默认安装，并且版本是v11.1
按你自己情况改。
改完之后，在系统变量里，PATH增加如下：
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11\lib
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11\libnvvp
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11\include

path_cuda.png

做完这些重启。
然后win+R进入cmd
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\extras\demo_suite

找到deviceQuery.exe,运行应出现result=Pass页面

devicequery.png

这里可以看到cuda toolKit版本是12.2，但运行库是11.1,所以可以运行tensorflow-gpu的。

找到bandwidthTest，运行它

bandwidth.png

也应该看到result=pass。
OK,这样可以认为你的开发环境的硬件驱动成功了。

软件部分

现在安装软件部分，就简单了，下载conda安装成功，启动虚拟环境。这里有一点python版本>3.7,小于3.11，GPT回答我3.11不支持，我没试，但我安装了3.8,然后安装
'''
pip install tensorflow-gpu==2.1.0
'''
网络环境有时不通，多试几次
安装成功后，按上面所说测试
'''
import tensorflow as tf
tf.test.is_gpu_available()
'''

tf_ture.png

至此成功，总结
1,windows只支持cuda tool kit 11.1
2,cuda tool kit对应版本是8.0.5
3,tensorflow安装版本小于2.1.0，安装时是
pip install tensorflow-gpu,如果不加GPU,默认装的是CPU版本，加了GPU,引用的还是tensorflow,这时启用了GPU
注意这二点，其他是常规操作。

如何在windows10上安装tensorflow-gpu