torch.cuda.get_arch_list

torch.cuda.get_arch_list() [source]

Return list CUDA architectures this library was compiled for.

Return type

list[str]