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]

On this page
Show Source
PyTorch Libraries