These snippets are in torch.utils.cpp_extension.py::_get_cuda_arch_flags. And the latest supported arch is sm_86.
PyTorch 1.13 released with CUDA 11.6 and 11.7, which did not support Hopper architectures. Update to any recent stable or nightly binary and it’ll work.
1 Like