广告
返回顶部
首页 > 资讯 > 后端开发 > Python >Tensorflow找不到GPU:tensorflow 2.11.0版本开始,在windows上不再支持GPU
  • 236
分享到

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

tensorflowwindowspython 2023-09-13 08:09:57 236人浏览 八月长安

Python 官方文档:入门教程 => 点击学习

摘要

背景 由于近期更换了电脑,所以需要在新电脑上重新配置Tensorflow的运行环境。本着使用最新版本的原则,按照官方指导教程进行安装以及配置CUDA, cuDNN等。此时我安装的最新版本为2.11.0

背景

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

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

import tensorflow as tfgpus = 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的问题已经解决。

来源地址:https://blog.csdn.net/bragen/article/details/129131278

--结束END--

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

本文链接: https://www.lsjlt.com/news/405578.html(转载时请注明来源链接)

有问题或投稿请发送至: 邮箱/279061341@qq.com    QQ/279061341

本篇文章演示代码以及资料文档资料下载

下载Word文档到电脑,方便收藏和打印~

下载Word文档
猜你喜欢
软考高级职称资格查询
编程网,编程工程师的家园,是目前国内优秀的开源技术社区之一,形成了由开源软件库、代码分享、资讯、协作翻译、讨论区和博客等几大频道内容,为IT开发者提供了一个发现、使用、并交流开源技术的平台。
  • 官方手机版

  • 微信公众号

  • 商务合作