【问题标题】:How to use GPU with keras如何在 keras 中使用 GPU
【发布时间】:2019-08-21 22:09:15
【问题描述】:

嘿,我在我的 mac 上使用 python3,想知道如何使用我的 GPU 更快地运行 Keras 神经网络。我尝试了以下方法,但它只会让情况变得更糟。

THEANO_FLAGS=device=gpu,floatX=float32 python program.py

【问题讨论】:

  • 为什么要使用tensorflow 标签,而您似乎使用的是 Theano 后端??

标签: python macos tensorflow keras gpu


【解决方案1】:

首先,检查您是否安装了 nvidia 工具包:nvidia-smi

您还需要与您的操作系统和 GPU 兼容的 cuda 驱动程序。运行以下命令检查是否已安装:nvcc -V

按照此处的说明安装cuda toolkit,其中还包括驱动器。

【讨论】:

    猜你喜欢
    • 2018-09-04
    • 2017-08-21
    • 1970-01-01
    • 1970-01-01
    • 2021-08-24
    • 1970-01-01
    • 2023-02-04
    • 1970-01-01
    • 1970-01-01
    相关资源
    最近更新 更多