tensorflow训练时如何指定使用哪个gpu

在使用 TensorFlow 进行训练时，可以使用以下代码指定使用哪个 GPU：

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"  # 使用编号为0的GPU

其中，`os.environ["CUDA_VISIBLE_DEVICES"]` 是一个环境变量，用于指定可见的 CUDA 设备。可以将其设置为一个 GPU 的编号，多个 GPU 的编号之间用逗号分隔。

例如，若要使用编号为 0 和 1 的两个 GPU 进行训练，可以使用以下代码：

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0,1"  # 使用编号为0和1的两个GPU

在指定完环境变量后，TensorFlow 会自动将计算任务分配到指定的 GPU 上进行运算。