如何在windows10上安装tensorflow-gpu
如何在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用以下命令检测：
'''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帐号，注册很麻烦，但安全性很高。
安装完全应该能看见环境变量时增加如图所示，
(查找环境变量方法，win+i,进入系统设置，到底部系统信息，高级系统设置,环境变量)，
如果安装成功，win+R进入cmd,输入nvidia-smi应该看到下述版本号：12.2
我的版本是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
做完这些重启。
然后win+R进入cmd
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\extras\demo_suite
找到deviceQuery.exe,运行应出现result=Pass页面
这里可以看到cuda toolKit版本是12.2，但运行库是11.1,所以可以运行tensorflow-gpu的。
找到bandwidthTest，运行它
也应该看到result=pass。
OK,这样可以认为你的开发环境的硬件驱动成功了。
软件部分
现在安装软件部分，就简单了，下载conda安装成功，启动虚拟环境。这里有一点python版本>3.7,小于3.11，GPT回答我3.11不支持，我没试试，但我安装了3.8,然后安装
'''shell
pip install tensorflow-gpu==2.1.0
'''
网络环境有时不通，多试几次
安装成功后，按上面所说测试
'''
import tensorflow as tf
tf.test.is_gpu_available()
'''
至此成功，总结 1,windows只支持cuda tool kit 11.1 2,cuda tool kit对应版本是8.0.5 注意这二点，其他是常规操作。
我苏格，跨境电商数据提供商。关注:SEO,反焦虑,内向