【问题标题】:tensorflow, tensorflow-gpu not recognizing the GPUtensorflow,tensorflow-gpu 无法识别 GPU
【发布时间】:2020-01-14 10:06:00
【问题描述】:

这可能是重复的帖子,但其他帖子都没有为我工作。 我有 tensorflow 1.14.0 和 tensorflow-gpu 1.14.0 我已经安装了所需的兼容 Cuda 和 Cudnn 版本,并在环境变量中指定了它们的路径。但仍然是 tensorflow 和/或 tensorflow-gpu,无法识别我的 GPU。 所以验证码的:

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']

OP:Windows 10

IDE:Jupyter (anaconda)

GPU 版本:GTX 960M

注意:有些帖子提到你必须在最后卸载 tensorflow,但是当我这样做时,上面的代码甚至不起作用......

【问题讨论】:

  • 请发布更多错误信息?你能运行 nvidia-smi 吗?

标签: python tensorflow gpu


【解决方案1】:

您绝对应该从系统中卸载 tensorflow,在该步骤之后代码无法正常工作表明您的 tensorflow-gpu 安装存在一些问题。这个accepted answer 建议通过 pycharms 卸载 tensorflow 并通过 pip 安装。

另外,如果你有机会,尝试升级到 tensorflow >= 2.0,Eager Execution 非常好,肯定会解决获取新驱动程序和兼容 Cuda 和 Cudnn 版本的麻烦。

【讨论】:

    猜你喜欢
    • 1970-01-01
    • 2023-03-12
    • 1970-01-01
    • 1970-01-01
    • 2021-10-24
    • 1970-01-01
    • 1970-01-01
    • 2021-12-20
    • 1970-01-01
    相关资源
    最近更新 更多