Sm_90a support?

1

I encounter this error when I install torch_geometric but I think it is likely an error of pytorch

File "/home/ubuntu/miniconda3/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1972, in <listcomp>
          supported_sm = [int(arch.split('_')[1])
      ValueError: invalid literal for int() with base 10: '90a

I found this is because

torch.cuda.get_arch_list()
>>> ['sm_50', 'sm_80', 'sm_86', 'sm_89', 'sm_90', 'sm_90a']

looks like sm_90a triggers this error.

2

Which PyTorch version are you using? I assumed the latest release already fixed this issue.

3

I use 2.5.1, which is the latest version?

4

Related issue: cpp_extension.py expects an integer on CUDA_ARCH, failing with Grace Hopper. · Issue #144037 · pytorch/pytorch · GitHub with my follow up.