首页 > 硬件设备 > GPU > RTX 5090 使用 PyTorch cu126 报错 no kernel image

RTX 5090 使用 PyTorch cu126 报错 no kernel image

GPU
## 一、问题背景
新笔记本到了,RTX 5090 的显卡,想着赶紧把 ComfyUI 跑起来做 AI 绘图。机器配置不错,硬件该装的都装了,接下来就是搭环境。
ComfyUI 安装没什么难度,git clone 完事。关键是 PyTorch 版本要对得上 CUDA 版本。
看了一眼显卡驱动:
CUDA 12.8,那 PyTorch 选 cu126(最新的稳定版)应该没问题吧?直接开下:
## 二、报错
下载花了挺久,2.6 GB 的包等了好一会儿。装完兴冲冲跑一下 ComfyUI,结果直接报错:
简单说就是:PyTorch 的 cu126 版本不带 RTX 5090 的驱动代码。
查了下 RTX 5090 用的是 Blackwell 架构,计算能力 sm_120。而 PyTorch 官方 cu126 的 wheel 包只编译到了 sm_89(Ada Lovelace 架构,也就是 RTX 4090 那代)。5090 是新架构,还没被包含进去。
## 三、解决过程
搜索得知 PyTorch 的 nightly 版本通常会提前加入对新架构的支持。需要换 cu128 nightly 版本:
2.6 GB 的文件才下完又得重下,先把 cu126 的 whl 删了释放空间。这次用 aria2 8 线程下载,加快速度:
装好以后跑 ComfyUI,这次顺利启动了。
## 四、为什么会这样
问题的根因是 PyTorch 的官方 wheel 包是预编译的,每个版本只包含特定计算能力(sm_xx)的内核镜像:
RTX 5090 的 sm_120 在 cu126 里没有被编译进去,所以报 "no kernel image" 错误。cu128 nightly 版本已经包含了 sm_120 的支持。
## 五、总结
首发了新一代显卡,在 AI 工具链上就要接受这种兼容性滞后。PyTorch 的 nightly 版本通常比稳定版早 1-3 个月支持新卡,如果你是首批 5090 用户:
1. 别下 cu126,直接上 cu128 nightly,省得白下 2.6 GB
2. 下载时加 --timeout 参数防超时
3. 等 PyTorch 稳定版更新后再切回来