Я хочу указать gpu для запуска моего процесса. И я установил его следующим образом:
import tensorflow as tf
with tf.device('/gpu:0'):
a = tf.constant(3.0)
with tf.Session() as sess:
while True:
print sess.run(a)
Однако он по-прежнему выделяет память в обоих моих двух gpus.
| 0 7479 C python 5437MiB
| 1 7479 C python 5437MiB