分步实操:5070TI+Stable Diffusion sm_120 报错解决流程
Stable Diffusion要求CUDA 11.8至12.x,若未安装或版本不符,需从NVIDIA开发者网站下载对应版本。若5070TI为笔误(实际应为4070 Ti或其他型号),需通过GPU-Z工具确认实际显卡的Compute Capability值。NVIDIA 40系显卡通常为sm_89,需相应调整启动参数。5070TI若为未发布型号,可能需手动指定计算能力。若驱动过旧,需到NVIDIA
检查显卡驱动兼容性
确保NVIDIA驱动版本支持CUDA 12.0及以上。通过NVIDIA控制面板或命令行工具nvidia-smi查看驱动版本。若驱动过旧,需到NVIDIA官网下载最新Game Ready或Studio驱动。
验证CUDA Toolkit安装
运行nvcc --version确认CUDA Toolkit版本。Stable Diffusion要求CUDA 11.8至12.x,若未安装或版本不符,需从NVIDIA开发者网站下载对应版本。安装时勾选"Visual Studio Integration"选项(Windows系统)。
配置PyTorch与CUDA匹配
在Python环境中执行以下命令验证PyTorch的CUDA支持:
import torch
print(torch.cuda.is_available())
print(torch.version.cuda)
若输出为False或版本不匹配,需通过pip重新安装对应版本:
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
修改Stable Diffusion启动参数
在启动命令中添加显式计算架构参数:
set COMMANDLINE_ARGS=--precision full --no-half --skip-torch-cuda-test
对于Linux系统:
export COMMANDLINE_ARGS="--precision full --no-half --skip-torch-cuda-test"
检查计算能力兼容性
5070TI若为未发布型号,可能需手动指定计算能力。编辑webui-user.bat(Windows)或webui.sh(Linux),在调用Python前添加环境变量:
set TORCH_COMMAND=install torch==2.1.0+cu121 torchvision==0.16.0+cu121 --index-url https://download.pytorch.org/whl/cu121
set PYTORCH_CUDA_ALLOC_CONF=max_split_size_mb:128
清理并重建虚拟环境
删除venv文件夹后重新启动WebUI,系统会自动重建Python环境。此操作可解决因依赖冲突导致的sm_120报错。
检查硬件实际参数
若5070TI为笔误(实际应为4070 Ti或其他型号),需通过GPU-Z工具确认实际显卡的Compute Capability值。NVIDIA 40系显卡通常为sm_89,需相应调整启动参数。
火山引擎开发者社区是火山引擎打造的AI技术生态平台,聚焦Agent与大模型开发,提供豆包系列模型(图像/视频/视觉)、智能分析与会话工具,并配套评测集、动手实验室及行业案例库。社区通过技术沙龙、挑战赛等活动促进开发者成长,新用户可领50万Tokens权益,助力构建智能应用。
更多推荐
所有评论(0)