python - cntk - theano vs tensorflow

¿Cómo puedo cambiar el dispositivo usado de theano?

Question

Intenté cambiar el dispositivo utilizado en el programa basado en theano.

from theano import config
config.device = "gpu1"

Sin embargo tengo error

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

Me pregunto cuál es la mejor manera de cambiar gpu a gpu1 en código.

Gracias

Answer 1

Elimine la configuración del "dispositivo" en .theanorc, luego en su código:

import theano.sandbox.cuda
theano.sandbox.cuda.use("gpu0")

Esto funciona para mi.

https://groups.google.com/forum/#!msg/theano-users/woPgxXCEMB4/l654PPpd5joJ

Answer 2

No hay manera de cambiar este valor en el código que se ejecuta en el mismo proceso. Lo mejor que podrías hacer es tener un proceso "padre" que altera, por ejemplo, la variable de entorno THEANO_FLAGS y genera hijos. Sin embargo, el método de desove determinará en qué entorno operan los niños.

Tenga en cuenta también que no hay manera de hacer esto de manera que mantenga la memoria de un proceso a través del cambio. No puede comenzar a ejecutar en la CPU, hacer un trabajo con los valores almacenados en la memoria, luego cambiar a la ejecución en la GPU y continuar utilizando los valores que aún están en la memoria desde la etapa anterior del trabajo (CPU). El proceso debe cerrarse y reiniciarse para que se aplique un cambio de dispositivo.

Tan pronto como import theano el dispositivo se arregla y no se puede cambiar dentro del proceso que hizo la importación.

Answer 3

Otra posibilidad que me funcionó fue establecer la variable de entorno en el proceso, antes de importar theano:

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