跳到主要内容
极客日志极客日志面向AI+效率的开发者社区
首页博客GitHub 精选镜像工具UI配色美学隐私政策关于联系
搜索内容 / 工具 / 仓库 / 镜像...⌘K搜索
注册
博客列表
PythonAI算法

Windows WSL 环境下 AMD 显卡部署 Stable Diffusion WebUI 与 ComfyUI

综述由AI生成在 Windows 系统中通过 WSL 2 结合 AMD 显卡部署 Stable Diffusion 的完整方案。内容包括安装 WSL 及 Ubuntu 系统、配置 AMD ROCm 驱动、搭建 Python 环境、安装 PyTorch ROCm 版本,以及最终部署 Stable Diffusion WebUI 和 ComfyUI。文中提供了详细的命令操作指南和依赖包下载地址,旨在帮助用户在本地实现 AI 图像生成工具的 GPU 加速运行。

莫名其妙发布于 2026/4/5更新于 2026/6/428 浏览

确保 Windows AMD 显卡驱动版本不低于 24.12.1,具体可参考官方说明。

一、安装 WSL 和 Ubuntu

  1. 安装 WSL2:
wsl --install
  1. 安装 Ubuntu(24.04、22.04 等):
wsl.exe --install ubuntu-24.04
  1. 更改 Ubuntu 安装位置(可选):
wsl --manage ubuntu-24.04 --move <location>
  1. 进入 WSL 实例:
# 输入 wsl -d <version> 进入指定版本或输入 wsl 进入默认实例
wsl -d ubuntu-24.04

可按 Ctrl+D 退出当前实例。

关闭实例:

wsl --shutdown

检查实例状态:

wsl -l -v

二、安装 ROCm(示例为安装 ROCm 6.3.4)

  1. 安装 AMD 统一驱动程序包存储库和安装程序脚本

具体 ROCm 适配情况可查看 AMD ROCm 兼容性文档。

注意:Ubuntu 24.04 和 Ubuntu 22.04 两者下载的包不一样,Ubuntu 24.04 对应 noble,Ubuntu 22.04 对应 jammy,具体下载网址为 AMD GPU Install Repo。

Ubuntu 24.04:

cd
sudo apt update
wget https://repo.radeon.com/amdgpu-install/6.3.4/ubuntu/noble/amdgpu-install_6.3.60304-1_all.deb
sudo apt install ./amdgpu-install_6.3.60304-1_all.deb

Ubuntu 22.04:

cd
sudo apt update
wget https://repo.radeon.com/amdgpu-install/6.3.4/ubuntu/jammy/amdgpu-install_6.3.60304-1_all.deb
sudo apt install ./amdgpu-install_6.3.60304-1_all.deb
  1. 查看可用用例列表(可选):
sudo amdgpu-install --list-usecase
  1. 安装 ROCm:
sudo amdgpu-install --usecase=graphics,opencl,video,hip
  1. 检查 ROCm 安装情况:
rocminfo

如果正确显示显卡信息,则安装成功。

三、安装 Conda(可选)

  1. 安装 Conda(用于管理 Python 版本):
cd
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
bash ./Miniconda3-latest-Linux-x86_64.sh

需重启控制台,只需要退出 WSL 就可以了,按 Ctrl+D,然后再启动 WSL 就可以了。

  1. 创建名为 sd 的环境,Python 版本为 3.10(建议 WebUI 和 ComfyUI 各自创建一个环境):
conda create -n sd python=3.10 -y
  1. 激活环境 sd:
conda activate sd

若需要退出当前 Conda 环境:

conda deactivate

四、安装 PyTorch(若没有使用 Conda,请自行安装 Python3)

  1. 下载对应 Python 版本的 whl 文件:
cd
wget https://repo.radeon.com/rocm/manylinux/rocm-rel-6.3.4/torch-2.4.0%2Brocm6.3.4.git7cecbf6d-cp310-cp310-linux_x86_64.whl
wget https://repo.radeon.com/rocm/manylinux/rocm-rel-6.3.4/torchvision-0.19.0%2Brocm6.3.4.gitfab84886-cp310-cp310-linux_x86_64.whl
wget https://repo.radeon.com/rocm/manylinux/rocm-rel-6.3.4/pytorch_triton_rocm-3.0.0%2Brocm6.3.4.git75cc27c2-cp310-cp310-linux_x86_64.whl
wget https://repo.radeon.com/rocm/manylinux/rocm-rel-6.3.4/torchaudio-2.4.0%2Brocm6.3.4.git69d40773-cp310-cp310-linux_x86_64.whl

(根据安装的 Python 版本确定安装包,安装包的名字中 cpxxx 为安装 Python 版本,例如 cp310 对应 Python 3.10 版本)

PyTorch ROCm 仓库

  1. 安装下载的 whl:
pip3 uninstall torch torchvision pytorch-triton-rocm
pip3 install torch-2.4.0%2Brocm6.3.4.git7cecbf6d-cp310-cp310-linux_x86_64.whl
pip3 install torchvision-0.19.0%2Brocm6.3.4.gitfab84886-cp310-cp310-linux_x86_64.whl
pip3 install torchaudio-2.4.0%2Brocm6.3.4.git69d40773-cp310-cp310-linux_x86_64.whl
pip3 install pytorch_triton_rocm-3.0.0%2Brocm6.3.4.git75cc27c2-cp310-cp310-linux_x86_64.whl
  1. 更新 WSL 兼容的运行时库:
location=$(pip show torch | grep Location | awk -F ": " '{print $2}')
cd ${location}/torch/lib/
rm libhsa-runtime64.so*
  1. libhsa-runtime64.so 至少需要安装 GCC 12.1,Conda 环境可以升级 gcc 版本(可选):
conda install -c conda-forge gcc=12.1.0
  1. 验证安装情况:
python3 -c 'import torch' 2> /dev/null && echo 'Success' || echo 'Failure'
python3 -c 'import torch; print(torch.cuda.is_available())'
python3 -c "import torch; print(f'device name [0]:', torch.cuda.get_device_name(0))"

输出结果分别为:Success、True、显卡型号则安装成功。

验证 PyTorch 环境:

python3 -m torch.utils.collect_env

五、安装 Stable-Diffusion-WebUI 和 ComfyUI

1. 安装 Stable-Diffusion-WebUI

(1)克隆 Stable Diffusion WebUI 到本地并安装依赖:

cd
git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
cd stable-diffusion-webui
pip install -r requirements.txt

(2)下载汉化包:

cd && cd ~/stable-diffusion-webui/extensions
git clone https://github.com/dtlnor/stable-diffusion-webui-localization-zh_CN.git

(3)运行 Stable-Diffusion-WebUI:

cd && cd stable-diffusion-webui
python launch.py

2. 安装 ComfyUI

(1)克隆 ComfyUI 到本地并安装依赖:

cd
git clone https://github.com/comfyanonymous/ComfyUI.git
cd ComfyUI
pip install -r requirements.txt

(2)与 Stable-Diffusion-WebUI 共享库(可选):

cd ~/ComfyUI
mv extra_model_paths.yaml.example extra_model_paths.yaml
vim extra_model_paths.yaml

在 vim 中将文件中 base_path: path/to/stable-diffusion-webui/ 改为 base_path: /home/用户名/stable-diffusion-webui/。

(3)运行 ComfyUI:

cd && cd ComfyUI
python main.py

目录

  1. 一、安装 WSL 和 Ubuntu
  2. 输入 wsl -d <version> 进入指定版本或输入 wsl 进入默认实例
  3. 二、安装 ROCm(示例为安装 ROCm 6.3.4)
  4. 三、安装 Conda(可选)
  5. 四、安装 PyTorch(若没有使用 Conda,请自行安装 Python3)
  6. 五、安装 Stable-Diffusion-WebUI 和 ComfyUI
  7. 1. 安装 Stable-Diffusion-WebUI
  8. 2. 安装 ComfyUI
  • 💰 8折买阿里云服务器限时8折了解详情
  • Magick API 一键接入全球大模型注册送1000万token查看
  • 🤖 一键搭建Deepseek满血版了解详情
  • 一键打造专属AI 智能体了解详情
极客日志微信公众号二维码

微信扫一扫,关注极客日志

微信公众号「极客日志V2」,在微信中扫描左侧二维码关注。展示文案:极客日志V2 zeeklog

更多推荐文章

查看全部
  • Python AI 大模型部署指南:本地运行、API 服务与 Docker 封装
  • OpenClaw 环境下的 Python 3.12 高性能编译指南
  • OpenClaw macOS 安装与配置指南
  • 朴素贝叶斯算法详解
  • Redis Hash 类型核心指令与实战详解
  • LangChain 技术解析:学习难点、版本演进与国产模型集成实践
  • 国产麒麟操作系统 V10 X86 架构离线安装 Nginx 及配置管理
  • C++ 异常处理机制:捕获、自定义与实战
  • 轻小说机翻机器人:开源日语小说翻译工具解析
  • Windows 系统下 Python 新一代包管理工具 uv 及 VSCode 配置
  • 拒绝经验过剩:程序员的工作只能是代码?
  • LLaMA-Factory 本地部署环境配置与 WebUI 启动指南
  • AI 提示词写作指南:精准表达与场景应用
  • Java 环境配置与基础概念详解
  • 基于 n8n 与代理服务构建自动化资讯采集推送系统
  • Python 基础语法详解与实战练习
  • Linux 进程优先级及 O(1) 调度机制解析
  • FlashTable 实测:AI 赋能低代码开发与企业级应用构建
  • Windows 本地部署 OpenClaw 并集成微信飞书
  • Pico 4XVR 1.10.13 安装包下载与安装教程

相关免费在线工具

  • 加密/解密文本

    使用加密算法(如AES、TripleDES、Rabbit或RC4)加密和解密文本明文。 在线工具,加密/解密文本在线工具,online

  • RSA密钥对生成器

    生成新的随机RSA私钥和公钥pem证书。 在线工具,RSA密钥对生成器在线工具,online

  • Mermaid 预览与可视化编辑

    基于 Mermaid.js 实时预览流程图、时序图等图表,支持源码编辑与即时渲染。 在线工具,Mermaid 预览与可视化编辑在线工具,online

  • 随机西班牙地址生成器

    随机生成西班牙地址(支持马德里、加泰罗尼亚、安达卢西亚、瓦伦西亚筛选),支持数量快捷选择、显示全部与下载。 在线工具,随机西班牙地址生成器在线工具,online

  • Gemini 图片去水印

    基于开源反向 Alpha 混合算法去除 Gemini/Nano Banana 图片水印,支持批量处理与下载。 在线工具,Gemini 图片去水印在线工具,online

  • curl 转代码

    解析常见 curl 参数并生成 fetch、axios、PHP curl 或 Python requests 示例代码。 在线工具,curl 转代码在线工具,online