WSL2 Ubuntu 快速部署 NovelAI (使用 NAIFU 前端)

写于 2022-10-22
AI 发展日新月异，如果你在 2023 年甚至更晚看到这篇文章，这个版本的 NAIFU 估计已经不用看了。

NAIFU 全称 NovelAI AI Image Frontend Ultimate，由 4chan 老哥基于 NovelAILeak 泄露文件魔改而成，界面与效果近似 NovelAI.net 官网。这里用的版本已捆绑模型文件，并支持 >75 词条。

和另一套前端（ Stable Diffusion web UI ）相比，NAIFU 操作更简单易上手，同时默认添加实用正反词条（“咒语”），新手更容易出高质量图片。但默认显存占用更大，有时候会无视词条、界面卡 BUG。建议新手先试用 NAIFU，感觉功能不够用了再换。

运行前提

  • Windows 10 20H1 (2004) 或更高版本，或 Win11，不限专业版或家庭版；

  • 在 BIOS／UEFI 中开启虚拟化相关功能（Intel VT-x、AMD SVM）；

  • 使用 NVIDIA 显卡，6GB 以上显存，近期显卡驱动。

快速安装 WSL2 Ubuntu

如果你之前没有安装过 WSL，只需一行命令即可：

使用管理员权限打开 Windows PowerShell
wsl --install

该命令会完成 WSL2 所需配置，并安装一个 Ubuntu 实例。

之后在开始菜单里找到 Ubuntu 并运行，按照提示设置用户名和密码。

基本环境

以下命令在 WSL2 Ubuntu (Bash) 中执行。

WSL2 Ubuntu 自带 N 卡 Linux 驱动，检查是否正常：

# 检查 Nvidia 显卡接入
nvidia-smi

安装必要的软件包

# 更新系统
sudo apt update && sudo apt upgrade -y && sudo apt autoremove

# 安装 Python 工具
sudo apt install -y python3-pip python3-virtualenv

# 安装下载工具 Aria2，如果你选择手动下载 NAIFU 包，可略过
apt install -y aria2

下载安装 NAIFU 整合包

这里使用 4GB 模型，而 7GB 模型完全没有必要，原因参考： https://stable-diffusion-book.vercel.app/4_model/#_3 
# 下载 NAIFU
aria2c --summary-interval=5 -x 3 --allow-overwrite=true -Z \
  https://pub-2fdef7a2969f43289c42ac5ae3412fd4.r2.dev/naifu.tar

# 解压
tar xf naifu.tar && rm naifu.tar

# 运行安装脚本（配置 Python 虚拟环境与依赖项）
cd naifu
bash ./setup.sh

# (可选)安装 xFormers 以提升性能，可能报错
# 建议等 v0.0.14 正式发布后再使用
./venv/bin/python -m pip install xformers

修改设置（可跳过）

# 直接调用 Windows 资源管理器
explorer.exe .

使用文本编辑器修改 run.sh

run.sh 文件修改内容
# 使用半精度浮点，适配 GeForce 显卡（16 系勿用）
export DTYPE=float16
# 保存所有生成的图片到 output 目录下
export SAVE_FILES=1
# 确保 WSL 的 CUDA 库可被调用
export LD_LIBRARY_PATH=/usr/lib/wsl/lib:$LD_LIBRARY_PATH

运行

# 确保 WSL 的 CUDA 库可被调用
export LD_LIBRARY_PATH=/usr/lib/wsl/lib:$LD_LIBRARY_PATH
# 运行
bash ./run.sh

打开浏览器，访问 http://localhost:6969 即可。

参考