Как использовать GPU с керасом

Question

Hy, я использую python3 на своем Mac и удивляюсь, как использовать мой графический процессор для более быстрой работы нейронных сетей Keras.Я попробовал фолловинг, но это только усугубило ситуацию.

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

Answer 1

Сначала проверьте, установлен ли у вас инструментарий nvidia с: nvidia-smi.

Вам также необходим драйвер cuda, совместимый с вашей ОС и графическим процессором. Проверьте, установлен ли он, запустив: nvcc -V.

Следуйте приведенным здесь инструкциям, чтобы установить cuda toolkit , который также включает в себя приводы.