可以使用TensorFlow中的 tf.device
函数来指定使用的GPU。例如,要将运算放在第二块GPU上,可以这样写:
with tf.device('/GPU:1'):
# 运算
或者,如果希望TensorFlow自动分配可用的GPU,可以使用 tf.device('/GPU:*')
。
注意:在使用GPU之前,需要确保已安装CUDA和TensorFlow GPU版本。
可以使用TensorFlow中的 tf.device
函数来指定使用的GPU。例如,要将运算放在第二块GPU上,可以这样写:
with tf.device('/GPU:1'):
# 运算
或者,如果希望TensorFlow自动分配可用的GPU,可以使用 tf.device('/GPU:*')
。
注意:在使用GPU之前,需要确保已安装CUDA和TensorFlow GPU版本。