欢迎光临散文网 会员登陆 & 注册

tensorflow gpu版最新安装说明

2023-03-23 20:19 作者:Yaxar_mbl  | 我要投稿

最近本人一直在使用CPU版的tensorflow跑神经网络,CPU的使用率总是飙到100%,风扇嗡嗡的转,烦死个人。但看独立显卡,使用率却一直为0%,明显在摸鱼偷懒。而且我也听说使用GPU跑神经网络有可能会快一点,于是我决定安装一个GPU版本的tensorflow。

安装这个GPU版的tensorflow真是折腾死人了,网上的教程大多都是旧的,有些已经不适用了。按照他们教的方法,整的我是云里雾里。最后终于在我自己的摸索下,找到一套可行的安装流程。

一、前言

在tensorflow英文的官方说明网站中可以看到图1的提示。这个提示只有英文版有,中文版没有,真的是坑死了,害我走了好多弯路。

图1 tensorflow官方提示(仅英文版有)

翻译过来的意思就是:Tensorflow 2.10是最后一个在本地windows上支持GPU的版本。从2.11版本开始,需要在windows WLS2(适用于 Linux 的 Windows 子系统)上安装才能使用GPU。

所以要在native-windows(一般人用的windows系统)上使用GPU运行tensorflow,就只能安装2.10版本及以下的版本,或者安装老版的tensorflow-gpu。

以上的部分内容引用自链接:https://blog.csdn.net/bragen/article/details/129131278

官方说明链接:https://tensorflow.google.cn/install/pip

本人使用的是Python3.9,而且我一般喜欢安装新版的软件,所以下面我将介绍在普通windows上可以安装的最新版本的tensorflow,也就是tensorflow2.10的安装方法。

二、安装CUDA

安装前先打开nvidia控制面板,点左下角的系统信息,再点击组件,看图2中圈中的序号,先记住,一会要用。

图2 nvidia控制面板

经过我的实验发现tensorflow2.10适配的11.x的CUDA。所以我选择安装CUDA11的最新版本11.8.0,如图3所示。

图3 CUDA下载界面

下载链接:https://developer.nvidia.com/cuda-toolkit-archive

安装前请一定确保安装的CUDA版本低于或等于图2中圈住的版本号。如不满足此要求请更新显卡驱动安装更低版本的CUDA

下载完后双击开始安装,选择自定义安装,只勾选图4中的这3项即可

图4 自定义安装

三、安装cuDNN

下载适配CUDA11版本的cuDNN,我选择安装最新版v8.8.1,如图5所示。

图5 cuDNN下载界面

下载链接:https://developer.nvidia.com/rdp/cudnn-download

下载完后是一个压缩包,解压后直接将binincludelib这三个文件夹复制到CUDA的安装目录,如图6-7所示。

图6 解压后内容
图7 CUDA安装目录

四、添加环境变量

此电脑→右键→属性→高级系统设置→环境变量→系统变量→Path→双击→检查以下路径是否在Path中,如果没有,就添加对应的路径,如图8所示。

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\bin

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\libnvvp

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\lib

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\include

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\extras\CUPTI\lib64

图8 添加环境变量

图8中环境变量的路径均E:\CUDA\开头,这是因为我安装CUDA时使用了自定义安装路径

官方说明链接:https://tensorflow.google.cn/install/gpu?hl=zh_cn#windows_setup

五、安装tensorflow

由于本人没有安装虚拟环境不推荐,不要学我),所以我只能介绍在系统直接上安装tensorflow的方法。

按下win键,输入cmd,在命令提示符处右键,选择以管理员身份运行

如果以前安装过其他版本的tensorflow,请在安装前务必卸载,可以使用以下命令:

tensorflow从2.x版本后已不再区分CPU版和GPU版,合为一个包就叫“tensorflow”,所以可以使用以下命令进行安装:

官方安装说明:https://tensorflow.google.cn/install/gpu?hl=zh_cn#pip_package

六、验证

可以使用以下代码检验是否安装成功

如果得到的结果不为0,则证明安装成功,可以使用GPU进行运算了。

官方说明链接:

https://tensorflow.google.cn/guide/gpu?hl=zh_cn#%E8%AE%BE%E7%BD%AE

七、其他

若在安装中出现缺少dll的问题,可以参考以下链接:

https://blog.csdn.net/weixin_44494790/article/details/110254621

但也有可能是CUDA安装的版本不对,可以根据缺少的dll名字下划线后面的数字来安装正确版本。

若安装中还出现了其他问题,可以留言评论(虽然我也不一定能解决😥大家可以一起探讨),或直接百度查询。

tensorflow gpu版最新安装说明的评论 (共 条)

分享到微博请遵守国家法律