How to install and run keras on GPU

原创于 2016-09-03 09:00:06 发布 · 637 阅读

CC 4.0 BY-SA版权

文章标签：

51 篇文章

订阅专栏

本文介绍如何在 Keras 中使用 GPU 加速深度学习模型训练，并提供了两种配置方法。第一种是在 Python 脚本中设置环境变量，第二种是通过命令行参数指定。文中还对比了 CPU 和 GPU 的运行效率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

install

import os
os.environ["THEANO_FLAGS"] = "mode=FAST_RUN,device=gpu,floatX=float32"
import theano
import keras

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

cpu方式运行:python mnist_cnn.py

gpu方式运行:THEANO_FLAGS=device=gpu,floatX=float32 python mnist_cnn.py
默认模式，即mode=FAST_COMPILE,gpu快的太多了！