2025-04-21 PyTorch 编译架构中没有 sm_89

发表于： 更新于： 分类于：
字数： 126 阅读：≈ 1分钟

今天发现很有意思的一件事情：配备 4090 显卡的服务器上安装 torch 之后，显示的架构并没有 sm_89，也没有 PTX（compute_xx）来支持 JIT。

(minimind) ➜  ~ python -c 'import torch;print(torch.cuda.get_arch_list())'
['sm_50', 'sm_60', 'sm_70', 'sm_75', 'sm_80', 'sm_86', 'sm_90']
(minimind) ➜  ~

sm_89 not listed in the torch.cuda.get_arch_list() - PyTorch Forums 这个讨论说明 sm_89 和 sm_86 是完全兼容的，除了 FP8 支持，这些 kernels 被专门处理。