torch.cuda.get_arch_list

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

返回此库编译所针对的CUDA架构列表。

返回类型

List[\str]