如何将GPU与keras结合使用

时间:2019-03-31 17:05:15

标签: python macos tensorflow keras gpu

嘿,我在Mac上使用python3,想知道如何使用我的GPU更快地运行Keras神经网络。我尝试了以下操作,但情况只会更糟。

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

1 个答案:

答案 0 :(得分:0)

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

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

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