嘿,我在Mac上使用python3,想知道如何使用我的GPU更快地运行Keras神经网络。我尝试了以下操作,但情况只会更糟。
THEANO_FLAGS=device=gpu,floatX=float32 python program.py
答案 0 :(得分:0)
首先,检查您是否安装了nvidia工具包:nvidia-smi
。
您还需要与您的OS和GPU兼容的cuda驱动程序。通过运行以下命令检查是否已安装:nvcc -V
。
按照此处的说明安装cuda toolkit,其中也包括驱动器。