Как сделать так, чтобы tenorflow и theano делили gpus одновременно?

Question

по некоторым сложным причинам я использую в своем коде python и tenorflow, и theano, и у меня есть 2 gpus, которыми я хочу поделиться, но, как указано в другой вопрос , есть некоторая проблема, я хочузнаете, есть ли какая-то хитрость для достижения этой цели, например, указание для использования tenorflow только 1 gpu, в то время как theano использует другое?

, а пока я могу отключить использование gpu theano только на os.environ['THEANO_FLAGS'] = 'device=cpu,floatX=float64', и разрешить tenorflow использовать все

os.environ['KERAS_BACKEND'] = 'theano'
os.environ['THEANO_FLAGS'] = 'device=cpu,floatX=float64'
import tensorflow as tf
import keras as ks

Answer 1

Я не пробовал это.Однако, если у вас несколько графических процессоров, вы можете принудительно запустить код на графическом процессоре, используя следующую хитрость:

import tensorflow as tf
with tf.device('/gpu:0'):
    # Run the tensorflow code

import tensorflow as tf
with tf.device('/gpu:1'):
    # Run the theano code

Надеюсь, это поможет!