在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帐号,注册很麻烦,但安全性很高。
安装完成应该能看见环境变量时增加如图所示,
(查找环境变量方法,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,然后安装
'''
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
3,tensorflow安装版本小于2.1.0,安装时是
pip install tensorflow-gpu,如果不加GPU,默认装的是CPU版本,加了GPU,引用的还是tensorflow,这时启用了GPU
注意这二点,其他是常规操作。
我苏格,跨境电商数据提供商。关注:SEO,反焦虑,内向,就是下楼前听下邻居有没有也准备下去