Windows10下Stable Diffusion环境Torch无法调用GPU问题求助
Windows10下Stable Diffusion Torch无法调用GPU的解决方案
强制指定webui使用CUDA
编辑webui-user.bat文件,找到set COMMANDLINE_ARGS=这一行,修改为:set COMMANDLINE_ARGS=--cuda保存后重新运行脚本,强制程序调用GPU而非CPU。
替换为webui兼容的Torch版本
不要自行安装官网最新版Torch,webui有适配的特定版本:- 打开命令提示符,进入Stable Diffusion webui根目录,执行卸载命令:
pip uninstall torch torchvision torchaudio -y - 根据你的CUDA版本手动安装适配Torch(以CUDA 11.8为例):
pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cu118 - 完成后直接运行
webui-user.bat,让脚本自动补全剩余依赖。
- 打开命令提示符,进入Stable Diffusion webui根目录,执行卸载命令:
检查CUDA环境变量配置
- 右键「此电脑」→「属性」→「高级系统设置」→「环境变量」
- 确认系统变量中存在
CUDA_PATH,值为你的CUDA安装目录(如C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8) - 将
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\bin和C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\libnvvp添加到系统Path变量中,重启电脑生效。
验证Torch对GPU的识别能力
打开命令提示符,执行以下Python代码:import torch print(torch.cuda.is_available()) print(torch.cuda.device_count()) print(torch.cuda.get_device_name(0))若输出
True、设备数≥1、显卡名称,则说明Torch已识别GPU,重新运行webui即可;若输出False,需确认你的NVIDIA显卡算力≥3.5,且显卡驱动版本符合CUDA的最低要求(可在NVIDIA官网查询对应CUDA版本的驱动版本)。排除集成显卡干扰
右键桌面→「NVIDIA控制面板」→「管理3D设置」→「全局设置」,选择「高性能NVIDIA处理器」,重启电脑后再运行webui。
内容的提问来源于stack exchange,提问作者bluefox
相关产品推荐
相关产品推荐

