Keras with theano на GPU

Question

При попытке запустить код Keras на графическом процессоре (CUDA), я не могу выполнить следующую инструкцию, как это было предложено во многих онлайн-ссылках.

set THEANO_FLAGS="mode=FAST_RUN,device=gpu,floatX=float32" & python theanogpu_example.py 

У меня возникла следующая ошибка.

ValueError: Invalid value ("FAST_RUN,device=gpu,floatX=float32") for configurati 
on variable "mode". Valid options are ('Mode', 'DebugMode', 'FAST_RUN', 'NanGuar 
dMode', 'FAST_COMPILE', 'DEBUG_MODE') 

Я пробовал использовать другой режим изнутри кода.

import theano 
theano.config.device = 'gpu' 
theano.config.floatX = 'float32' 

Я получаю следующее сообщение об ошибке.

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

Помимо зная, как заставить ее работать, я бы также воспользоваться этой возможностью, чтобы задать простой вопрос. Как узнать в Windows, что такое мое устройство, то есть «gpu» или «gpu1» или «gpu0»? Я пробовал все 3 для моего случая, но это не дало результата.

Любые предложения будут оценены.

Answer 1

Лучший способ использует THEANO_FLAGS до того кода запуска, поскольку переменные конфигурации не может быть изменен после импорта Theano, попробуйте следующее:

import os 
os.environ['THEANO_FLAGS'] = "device=cuda,force_device=True,floatX=float32" 
import theano