comfyUI 運行出錯 FATAL: kernel fmha_cutlassF_f32_aligned_64x64_rf_sm80 is for sm80-sm100, but was built for sm37

在公司的圖形機器上運行 ComfyUI + WAN 時出現下面問題

 FATAL: kernel fmha_cutlassF_f32_aligned_64x64_rf_sm80 is for sm80-sm100, but was built for sm37

算圖機使用的 torch 版本如下 torch 2.7.1+cu118 torchaudio 2.7.1+cu118 torchsde 0.2.6 torchvision 0.22.1+cu118

檢查 cuda 版本

GeForce RTX 5090

CUDA Version: 13.0

從官網看 RTX 5090 支持 12.0 版本 CUDA

這是因爲本地的 pytorch 版本太舊

到官網查詢版本

https://pytorch.org/get-started/locally/




$ pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu126




查詢目前版本




$ pip list | findstr torch

使用這個版本后出現下面問題

後面查詢 no kernel image 問題其實還是 pytorch 並不支持 sm_120(Blackwell架構)

查看后發現并沒有支持 sm_120

CUDA architectures: ['sm_37', 'sm_50', 'sm_60', 'sm_61', 'sm_70', 'sm_75', 'sm_80', 'sm_86', 'sm_90', 'compute_37']

sm_50, sm_52, sm_53: Maxwell (GTX 900 系列)

sm_60, sm_61, sm_62: Pascal (GTX 1000, P100)

sm_70, sm_72, sm_75: Volta/Turing (V100, RTX 2000, GTX 1600)

sm_80, sm_86, sm_87: Ampere (A100, RTX 3000)

sm_89: Ada Lovelace (RTX 4000)

sm_90, sm_90a: Hopper (H100, H200)

sm_100: Blackwell (RTX 5090, B100) 

重新安裝 nightly 版本 cu130

$ pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130

後面運行就正常了