torch.cuda.device_count()

torch.cuda.device_count()[源代码]

返回可用GPU的数量。

返回类型

int