linux环境安装autokeras,Keras下GPU/CPU模式切换

If you are running on the TensorFlow backend, your code will automatically run on GPU if any available GPU is detected. If you are running on the Theano backend, you can use one of the following methods:

Method 1: use Theano flags.

THEANO_FLAGS=device=gpu,floatX=float32 Python my_keras_script.py

The name ‘gpu’ might have to be changed depending on your device’s identifier (e.g. gpu0, gpu1, etc).

Method 2: set up your .theanorc: Instructions

sudo vim ~/.theanorc

add these content

[global]

device=gpu

floatX=float32

Method 3: manually set theano.config.device, theano.config.floatX at the beginning of your code:

import theano

theano.config.device = ‘gpu’

theano.config.floatX = ‘float32’