如何同时使Tensorflow和Theano共享GPU

Question

出于一些复杂的原因，我在python代码中同时使用Tensorflow和Theano，并且我有2个GPU，我希望它们共享，但是如另一个问题，我有一些问题，我想知道是否有一些技巧可以实现这一目标，例如指定TensorFlow在Theano使用另一个时仅使用1 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

希望这会有所帮助！