如何知道哪个GPU可用

您可以使用以下代码来确定可用的GPU：

import tensorflow as tf

gpus = tf.config.experimental.list_physical_devices('GPU')
if gpus:
    for gpu in gpus:
        print("GPU可用:", gpu)
else:
    print("未找到可用的GPU")

这将打印出所有可用的GPU设备。如果没有找到可用的GPU，将会显示"未找到可用的GPU"。