torch.cuda.get_arch_list

torch.cuda.get_arch_list()[source][source]

返回该库编译支持的 CUDA 架构列表。

返回类型:

list[str]