Rate this Page

torch.cuda.get_arch_list#

torch.cuda.get_arch_list()[source]#

Return list CUDA architectures this library was compiled for.

Return type

list[str]