兼容性分析
NVIDIA GeForce RTX 5070 Ti 尚未发布,目前该型号不存在。若用户实际指的是 RTX 4070 Ti 或其他型号,需注意 SM(Streaming Multiprocessor)架构版本差异。Stable Diffusion 要求 CUDA 计算能力 6.0 及以上,常见错误 sm_120 is not compatible 通常由以下原因引起:
- 驱动/CUDA 工具包版本过旧:旧版工具包可能不支持新显卡的 SM 架构。
- PyTorch 版本不匹配:PyTorch 未正确识别显卡计算能力。
- 自定义编译错误:手动编译时指定了错误的
-gencode参数。
解决方案
更新驱动和CUDA工具包 访问 NVIDIA 官方网站下载最新驱动,并安装与显卡匹配的 CUDA 12.x 版本。确保环境变量 PATH 包含 CUDA 的 bin 和 lib 路径。
验证 PyTorch 版本 使用以下命令安装支持 Ampere 架构(如 RTX 40 系列)的 PyTorch:
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
强制指定计算能力 若需从源码编译,在 CMake 或 nvcc 命令中明确指定计算能力(如 RTX 4070 Ti 为 sm_89):
export TORCH_CUDA_ARCH_LIST="8.9"
环境检查步骤
运行以下 Python 代码验证环境是否正常:
import torch
print(torch.__version__) # 应 ≥2.0
print(torch.cuda.is_available()) # 应返回 True
print(torch.cuda.get_device_capability()) # 40系显卡应显示 (8,9)
若仍报错,建议彻底卸载原有驱动和 CUDA 后重新安装,或使用 Docker 镜像(如 nvcr.io/nvidia/pytorch:23.10-py3)规避环境冲突。
5684

被折叠的 条评论
为什么被折叠?



