8wDlpd.png
8wDFp9.png
8wDEOx.png
8wDMfH.png
8wDKte.png

尽管设置正确,但仍无法在 GPU 模式下运行 TensorFlow

Sathish Manohar 2月前

103 0

问题:我一直尝试在 GPU 模式下运行 TensorFlow,但没有成功。当我运行以下代码来检查可用设备时:from tensorflow.python.client import device_libdef

问题:我一直尝试在 GPU 模式下运行 TensorFlow,但没有成功。当我运行以下代码来检查可用设备时:

from tensorflow.python.client import device_lib

def get_available_devices():
    local_device_protos = device_lib.list_local_devices()
    return [x.name for x in local_device_protos]

print(get_available_devices())

输出为:

['/device:CPU:0']

预期输出应该是:

['/device:CPU:0', '/device:GPU:0']

当我跑步时:

import tensorflow as tf

print("Num GPUs Available: ", len(tf.config.list_physical_devices('GPU')))

输出为:

Num GPUs Available:  0

以下是我的规格:

  • CUDA 版本:12.2
  • cuDNN 版本:8.9.2
  • GPU型号:Nvidia GeForce MX350
  • 操作系统:Windows 11
  • TensorFlow 版本:2.16.0

我已采取的步骤:

  1. 已安装 CUDA 12.2
  2. 下载并设置 cuDNN 8.9.2
  3. 在环境变量中设置CUDA路径
  4. 重启电脑

这里可能存在什么问题?我该如何解决它以确保 TensorFlow 识别并使用 GPU?

帖子版权声明 1、本帖标题:尽管设置正确,但仍无法在 GPU 模式下运行 TensorFlow
    本站网址:http://xjnalaquan.com/
2、本网站的资源部分来源于网络,如有侵权,请联系站长进行删除处理。
3、会员发帖仅代表会员个人观点,并不代表本站赞同其观点和对其真实性负责。
4、本站一律禁止以任何方式发布或转载任何违法的相关信息,访客发现请向站长举报
5、站长邮箱:yeweds@126.com 除非注明,本帖由Sathish Manohar在本站《tensorflow》版块原创发布, 转载请注明出处!
最新回复 (0)
  • 我还检查了 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.2\include 中的 cudnn_version.h

返回
作者最近主题: