深度学习如何指定GPU

可以使用TensorFlow中的 tf.device 函数来指定使用的GPU。例如，要将运算放在第二块GPU上，可以这样写：

with tf.device('/GPU:1'):
  # 运算

或者，如果希望TensorFlow自动分配可用的GPU，可以使用 tf.device('/GPU:*') 。

注意：在使用GPU之前，需要确保已安装CUDA和TensorFlow GPU版本。