How can I change device used of theano

Question

I tried to change the device used in theano-based program.

from theano import config
config.device = "gpu1"

However I got error

Exception: Can't change the value of this config parameter after initialization!

I wonder what is the best way of change gpu to gpu1 in code ?

Thanks

Answer 1

Another possibility which worked for me was setting the environment variable in the process, before importing theano:

import os    
os.environ['THEANO_FLAGS'] = "device=gpu1"    
import theano