当前位置: 代码网 > 科技>操作系统>Windows > Tensorflow找不到GPU:tensorflow 2.11.0版本开始,在windows上不再支持GPU

Tensorflow找不到GPU:tensorflow 2.11.0版本开始,在windows上不再支持GPU

2024年08月01日 Windows 我要评论
Tensorflow从2.11.0版本开始,在native-windows上不再支持GPU。

背景

由于近期更换了电脑,所以需要在新电脑上重新配置tensorflow的运行环境。本着使用最新版本的原则,按照官方指导教程进行安装以及配置cuda, cudnn等。此时我安装的最新版本为2.11.0,经测试tensorflow无法使用gpu。

测试是否能够使用gpu的代码:

import tensorflow as tf
gpus = tf.config.experimental.list_physical_devices(device_type='gpu')
cpus = tf.config.experimental.list_physical_devices(device_type='cpu')
print(gpus, cpus)

程序运行结果为:

[] [physicaldevice(name='/physical_device:cpu:0', device_type='cpu')]

显然tensorflow没有找到gpu,只有cpu。

以下是官网给出的安装教程:
官方链接: 使用 pip 安装 tensorflow

  1. 在虚拟环境中使用pip安装tensorflow
pip install tensorflow
  1. 接着配置相关软件
  1. 配置环境变量
    确保安装的 nvidia 软件包与上面列出的版本一致。特别是,如果没有 cudnn64_8.dll 文件,tensorflow 将无法加载。如需使用其他版本,请参阅在 windows 下从源代码构建指南。将 cuda®、cupti 和 cudnn 安装目录添加到 %path% 环境变量中。例如,如果 cuda® 工具包安装到 c:\program files\nvidia gpu computing toolkit\cuda\v11.0,并且 cudnn 安装到 c:\tools\cuda,请更新 %path% 以匹配路径:
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\lib64;%path%
set path=c:\program files\nvidia gpu computing toolkit\cuda\v11.0\include;%path%
set path=c:\tools\cuda\bin;%path%

问题解决

tensorflow版本从2.x开始不再区分cpu版和gpu版,因此在软件配置正确的情况下,应该是可以找到gpu设备的。我反复按照官网教程确认了各软件的版本以及环境变量等均没有错误,但问题仍没有解决。

我将官网的语言调到英文时,发现了这样一条提示(在中文版的网页中是没有的):
2.11.0版本不再支持gpu
根据该条提示,tensorflow 2.10是最后一个在本地windows上支持gpu的版本。从2.11版本开始,需要在windows wls2(适用于 linux 的 windows 子系统)上安装才能使用gpu。所以要在native-windows上使用gpu,就只能安装2.10.0版本及以下的版本,或者安装老版的tensorflow-gpu。

所以我卸载了之前安装的2.11.0版本,安装了2.10.0版本:

pip install tensorflow==2.10.0

然后测试是否能够使用gpu:
找到gpu
可见,找不到gpu的问题已经解决。

(0)

相关文章:

版权声明:本文内容由互联网用户贡献,该文观点仅代表作者本人。本站仅提供信息存储服务,不拥有所有权,不承担相关法律责任。 如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 2386932994@qq.com 举报,一经查实将立刻删除。

发表评论

验证码:
Copyright © 2017-2025  代码网 保留所有权利. 粤ICP备2024248653号
站长QQ:2386932994 | 联系邮箱:2386932994@qq.com