keras配置theano/tensorflow GPU

GPU的cuda环境自行配置

Anoconda 自行安装

创建conda环境

conda create -n keras python=3.5

安装theano

conda install theano

安装tensorflow-gpu

conda install tensorflow-gpu

配置keras的backen，keras默认的backen是tensorflow

vim ~/.keras/keras.json

{
“image_dim_ordering”: “th”,
“epsilon”: 1e-07,
“floatx”: “float32”,
“backend”: “tensorflow”
}
把上面 “backend”: "tensorflow"改成 “backend”: “theano”，这样就可以了

配置theano GPU

#生成配置文件
vim ~/.theanorc

[global]
device=gpu
floatX=float32
root=/usr/local/cuda
[nvcc]
fastmath = True
[blas]
ldflags = -lopenblas
[cuda]
root = /usr/local/cuda
[lib]
cnmem=0.8

输入上面内容即可。
在命令行输入 import theano，显示使用的GPU设备表示GPU配置成功。

测试用例：

from keras.models import Sequential
from keras.layers.core import Dense, Dropout, Flatten
from keras.layers.convolutional import Convolution2D, MaxPooling2D
model.add(Convolution2D(32, 3, 3)
model.add(Activation(‘relu’))
model.add(MaxPooling2D(pool_size=(2, 2))
model.add(Dropout(0.25))
model.add(Dense(10))
model.add(Activation(‘softmax’))