检查显卡驱动版本

确保NVIDIA显卡驱动版本支持CUDA 11.8及以上。访问NVIDIA官网下载最新驱动，或通过GeForce Experience更新。运行nvidia-smi命令确认驱动版本和CUDA版本兼容性。

重新配置CUDA环境

卸载现有CUDA工具包，从NVIDIA官网下载CUDA 11.8版本并安装。安装时需勾选"NSight"和"Visual Studio集成"组件。完成后通过nvcc --version验证安装是否成功。

修改Stable Diffusion启动参数

在启动脚本中添加显式指定计算能力的参数。例如，将原命令改为：

python launch.py --precision full --no-half --skip-torch-cuda-test --disable-nan-check --opt-sdp-attention --xformers

若使用WebUI，在webui-user.bat中添加：

set COMMANDLINE_ARGS=--xformers --no-half

验证显卡计算能力

5070TI的计算能力为8.6（Ampere架构），需确保PyTorch版本支持。运行以下代码测试：

import torch
print(torch.cuda.get_device_capability())

若输出非(8,6)，需重新安装匹配的PyTorch版本：

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

清理并重建虚拟环境

删除原有虚拟环境文件夹，新建环境后重新安装依赖：

conda create -n sd python=3.10.6
conda activate sd
pip install -r requirements.txt

检查xFormers兼容性

手动编译xFormers以适配Ampere架构：

pip install ninja
pip install -v -U git+https://github.com/facebookresearch/xformers.git@main#egg=xformers

编译时添加环境变量：

export TORCH_CUDA_ARCH_LIST="8.6"