如何使tensorflow和theano同时共享GPU？

Question

出于某些复杂的原因，我在我的python代码中同时使用了tensorflow和theano，并且我有两个要共享的GPU，但是正如another question中所述，我想知道是否存在一些技巧来实现这一点，例如指定tensorflow只使用1 gpu，而theano使用另一个gpu？

现在我只能通过os.environ['THEANO_FLAGS'] = 'device=cpu,floatX=float64'禁用theano的gpu使用，而让tensorflow使用所有

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

Answer 1

我还没有尝试过。但是，如果您有多个GPU，则可以使用以下技巧强制在GPU上运行代码：

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

希望这会有所帮助！