python - 如何让tensorflow和theano同时共享gpus？

Question

由于一些复杂的原因，我在我的 python 代码中同时使用了 tensorflow 和 theano，并且我有 2 个 gpus 我希望它们共享，但是正如另一个问题中所述，存在一些问题，我想知道是否有一些技巧可以实现这一点，就像指定 tensorflow 只使用 1 个 gpu 而 theano 使用另一个？

现在我只能禁用 theano 的 gpu 使用os.environ['THEANO_FLAGS'] = 'device=cpu,floatX=float64'，并让 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

希望这可以帮助！