如何在 keras 中使用 GPU

Question

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

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

Answer 1

首先，检查您是否安装了 nvidia 工具包： nvidia-smi 。

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

按照此处的说明安装还包括驱动器的cuda 工具包。