【问题标题】:TensorFlow not detecting GPU (Dell Vostro with NVIDIA GeForce GTX 1060)TensorFlow 未检测到 GPU(Dell Vostro 与 NVIDIA GeForce GTX 1060)
【发布时间】:2021-08-28 20:58:03
【问题描述】:

操作系统:Windows 10(也尝试使用 Ubuntu 无济于事)
CPU:英特尔酷睿 i7-8750H @ 2.2 GHz
GPU:采用 Max-Q 设计的 NVIDIA GeForce GTX 1060
Python 3.8 版
TensorFlow 2.3.0 版(我确实使用tensorflow-gpu 安装了它)

代码:

import tensorflow as tf
print("Num GPUs Available: ", len(tf.config.list_physical_devices('GPU')))
print("Num CPUs Available: ", len(tf.config.list_physical_devices('CPU')))

结果:

Num GPUs Available:  0
Num CPUs Available:  1

有什么想法吗?

【问题讨论】:

  • 你安装了CUDA吗?
  • 如果您正在进行 conda 安装,anaconda 将为您获取 cudnn 和 cuda 工具包依赖项并将其添加到路径中。如果您使用 pip 安装,则必须手动获取所有依赖项。如果您正在使用 tensorflow-gpu,我强烈建议您使用 anaconda。使用方法如下:towardsdatascience.com/…
  • 是的,我已经安装了 CUDA
  • 您能告诉我们安装的 CUDA 和 cuDNN 的版本吗?
  • 我升级到 Python 3.9.6,现在它可以工作了。谢谢

标签: python tensorflow gpu


【解决方案1】:

来自评论

升级到 Python 3.9.6 后问题已解决(转述自 伊丹·托比斯)

为了社区的利益,提供经过测试的构建配置

【讨论】:

    猜你喜欢
    • 1970-01-01
    • 1970-01-01
    • 2021-08-10
    • 1970-01-01
    • 2017-09-15
    • 2019-02-15
    • 1970-01-01
    • 2018-05-31
    • 2020-12-10
    相关资源
    最近更新 更多