当前位置:
首页 > AI笔记 > ComfyUI部署问题:50系显卡安装torch+torchvision出现不兼容sm_120算力架构

ComfyUI部署问题:50系显卡安装torch+torchvision出现不兼容sm_120算力架构

  今天本来想玩玩ComfyUI做AI短时频的,可是在部署本地ComfyUI中遇到几个问题,跟大家说说。我一开始安装的PyTorch 1.12+cu126安装完成我测试输出环境,提示兼容问题不支持sm_120,问了豆包告诉我是50系显卡太新了,老的版本PyTorch还没完善。然后我就开始寻找教程...

ComfyUI部署问题:50系显卡安装torch+torchvision出现不兼容sm_120算力架构

  下面我来总结一下:

  后来我安装了2.6.0+cu126提示PyTorch 2.6.0+cu126 Windows 官方预编译包本身就没有编译 sm_120(RTX 50 系专属算力),这是 PyTorch 官方编译策略问题:

  Windows 平台下,针对RTX50系列sm_120的原生支持,仅在cu132(CUDA13.2)及以上编译包中提供;所有 cu126 版本(2.6/2.7/2.8…)Windows 包都不带sm_120,所以警告会一直存在。

  最后终于在我重复安装测试下找到了一个兼容sm_120架构的组合:

  先卸载当前 cu126 旧版本

  PowerShell 执行:

  pip uninstall torch torchvision torchaudio -y

  安装torch 2.12.0+cu132 + torchvision 0.27.0+cu132完整匹配版本,后面我们在补齐torchaudio安装包。

  PowerShell 执行:

  pip install torch==2.12.0+cu132 torchvision==0.27.0+cu132 --index-url https://download.pytorch.org/whl/cu132 --timeout 1800 --retries 20 --force-reinstall

  测试结果:

  ===============

  torch 版本: 2.12.0+cu132

  torchvision 版本: 0.27.0+cu132

  CUDA 可用: True CUDA

  运行版本: 13.2

  显卡算力架构列表: ['sm_75', 'sm_80', 'sm_86', 'sm_90', 'sm_100', 'sm_120']

  =================

  关键校验结果解读: PyTorch 2.12.0+cu132 + TorchVision 0.27.0+cu132 版本匹配正常算力列表已包含 sm_120,原生完美适配 RTX 5060 Ti,彻底消除了之前的兼容警告双卡显卡、CUDA 加速全部识别正常,CUDA 13.2 向下兼容你本机 12.6 驱动,无任何风险。

  最后还要补齐安装torchaudio,不安装的话,启动ComfyUI还是后提示错误的。

  powershell 执行:

  pip install torchaudio

  安装完成后,所有组件都已完美安装:

  torch: 2.12.0+cu132 (CUDA 13.2 版本)

  torchvision: 成功升级为 0.27.0+cu132

  torchaudio: 成功安装 2.11.0+cu132

  经我测试torch: 2.12+cu132才可以完美过sm_120算力架构!现在万事俱备,你可以直接去启动 ComfyUI 了。

ComfyUI部署问题:50系显卡安装torch+torchvision出现不兼容sm_120算力架构:等您坐沙发呢!

发表评论

表情
还能输入210个字