当前位置：首页 > news >正文

AI头像生成器部署教程（Windows WSL2）：Ubuntu子系统运行Qwen3-32B+Gradio全记录

news 2026/7/28 15:47:42

AI头像生成器部署教程（Windows WSL2）：Ubuntu子系统运行Qwen3-32B+Gradio全记录

想给自己设计一个独一无二的头像，但苦于没有绘画功底？想让AI帮你把脑海中的形象画出来，却不知道该怎么描述？今天，我来分享一个超实用的解决方案——在Windows电脑上，通过WSL2运行一个AI头像生成器。

这个工具基于强大的Qwen3-32B大语言模型，你只需要用简单的语言描述想要的风格，它就能帮你生成一份详细到头发丝的头像设计文案。这份文案可以直接复制粘贴到Midjourney、Stable Diffusion这些AI绘图工具里，瞬间把你的想法变成精美的图片。

无论你是想做一个赛博朋克风的虚拟形象，还是想要一个古风动漫角色，这个工具都能帮你搞定。接下来，我会手把手带你完成从零到一的完整部署过程，让你在自己的电脑上就能拥有一个专属的头像创意助手。

1. 环境准备：搭建你的AI创作空间

在开始之前，我们先来了解一下需要准备什么。整个过程就像搭积木，我们把几个关键的组件组合起来，就能构建出完整的系统。

1.1 你需要准备的东西

首先，确保你的电脑满足以下基本要求：

操作系统：Windows 10 版本 2004 或更高版本（包括 Windows 11）
内存：建议 16GB 或以上（运行大模型比较吃内存）
存储空间：至少需要 50GB 可用空间（模型文件比较大）
网络：稳定的网络连接（下载镜像和模型需要时间）

如果你不确定自己的Windows版本，可以按Win + R键，输入winver然后回车，就能看到详细的版本信息。

1.2 了解技术组件

我们这次部署用到了几个关键技术，我用大白话给你解释一下：

WSL2（Windows Subsystem for Linux 2）：可以理解为在Windows里运行的一个“Linux虚拟机”，让我们能在Windows上使用Linux环境。
Ubuntu：一个很流行的Linux操作系统，我们会在WSL2里安装它。
Qwen3-32B：阿里通义千问的一个大语言模型，有320亿参数，理解能力和创意都很强。
Ollama：一个专门用来在本地运行大模型的工具，让部署变得很简单。
Gradio：一个Python库，能快速为AI模型创建网页界面，不用写复杂的网页代码。

整个流程就是：在Windows上启用WSL2 → 安装Ubuntu → 在Ubuntu里安装Ollama → 用Ollama拉取Qwen3-32B模型 → 用Gradio创建网页界面。

听起来步骤不少，但别担心，我会带着你一步步走，每个步骤都有详细的命令和说明。

2. 第一步：安装和配置WSL2与Ubuntu

这是整个部署的基础，就像盖房子要先打好地基。WSL2让我们能在Windows上运行Linux系统，而Ubuntu是我们选择的Linux发行版。

2.1 启用WSL2功能

首先，我们需要打开Windows的WSL功能。有图形界面和命令行两种方法，我推荐用命令行，更快更直接。

按Win + X键，选择“Windows终端（管理员）”或“命令提示符（管理员）”。在打开的窗口里，输入以下命令：

wsl --install

这个命令会自动完成几件事：

启用“适用于Linux的Windows子系统”可选功能
启用“虚拟机平台”可选功能
下载并安装最新的Linux内核更新包
将WSL2设置为默认版本
安装Ubuntu发行版（默认的）

执行完成后，系统会提示你重启电脑。一定要重启，否则更改不会生效。

重启后，再次打开命令行窗口，输入以下命令检查WSL版本：

wsl --version

如果看到类似下面的输出，说明安装成功：

WSL 版本： 2.0.9.0 内核版本： 5.15.133.1-1 WSLg 版本： 1.0.59 MSRDC 版本： 1.2.4672 Direct3D 版本： 1.611.1-81528511 DXCore 版本： 10.0.25131.1002-220531-1700.rs-onecore-base2-hyp Windows 版本： 10.0.22631

2.2 设置Ubuntu系统

重启后，你会在开始菜单里看到Ubuntu的图标，点击它就会启动Ubuntu的安装过程。第一次启动需要一些时间，系统会下载必要的文件。

安装过程中，系统会提示你：

创建用户名（不能用大写字母）
设置密码（输入时不会显示字符，正常输入即可）

设置完成后，你就进入了Ubuntu的命令行界面。为了后续操作顺利，我们先更新一下系统软件包：

sudo apt update && sudo apt upgrade -y

这个命令会更新软件包列表并升级所有可升级的软件包。sudo表示以管理员权限运行，apt是Ubuntu的包管理工具，update是更新列表，upgrade是升级软件，-y表示自动回答“是”来确认操作。

更新过程可能需要几分钟，取决于你的网络速度。完成后，我们的基础环境就准备好了。

3. 第二步：安装和配置Ollama

Ollama是我们运行大模型的关键工具，它让本地部署大模型变得像安装普通软件一样简单。

3.1 安装Ollama

在Ubuntu终端中，输入以下命令来安装Ollama：

curl -fsSL https://ollama.com/install.sh | sh

这个命令做了什么呢？

curl是一个从网络下载文件的工具
-fsSL是几个参数组合：-f（失败时静默）、-s（静默模式）、-S（显示错误）、-L（跟随重定向）
后面的网址是Ollama的安装脚本
| sh表示把下载的内容通过管道传递给sh命令执行

安装过程会自动下载Ollama并设置好系统服务。安装完成后，启动Ollama服务：

ollama serve

第一次运行可能会提示你Ollama服务正在启动。为了确保Ollama在后台持续运行，我们可以把它设置为系统服务：

sudo systemctl enable ollama sudo systemctl start ollama

检查服务状态，确认一切正常：

sudo systemctl status ollama

如果看到“active (running)”的字样，说明Ollama服务已经在正常运行了。

3.2 拉取Qwen3-32B模型

现在我们来获取最重要的部分——Qwen3-32B模型。这个模型有320亿参数，能力很强，但文件也比较大（大约60GB），所以需要一些下载时间。

打开一个新的终端窗口（让Ollama服务在后台继续运行），输入：

ollama pull qwen2.5:32b

这里解释一下：

ollama pull是拉取模型的命令
qwen2.5:32b指定了我们要拉取的模型名称和版本
32b表示32B参数版本，也就是320亿参数

下载时间取决于你的网速，可能需要几个小时。你可以看到下载进度，模型会分成多个部分（称为“层”）下载。下载过程中，你可以去做其他事情，不用一直盯着。

下载完成后，验证一下模型是否可用：

ollama list

这个命令会列出所有已下载的模型，你应该能看到qwen2.5:32b在列表中。

为了测试模型是否能正常工作，我们可以运行一个简单的对话：

ollama run qwen2.5:32b

在出现的提示符后，输入“你好”，看看模型是否会回应。如果能看到中文回复，说明模型运行正常。按Ctrl+D可以退出对话。

4. 第三步：创建AI头像生成器应用

模型准备好了，现在我们来创建实际可用的头像生成器应用。我们会用Python编写一个简单的程序，然后通过Gradio给它加上网页界面。

4.1 准备Python环境

首先，确保Ubuntu里安装了Python和必要的工具：

sudo apt install python3 python3-pip python3-venv -y

然后创建一个专门的项目目录，并进入该目录：

mkdir ~/ai-avatar-generator && cd ~/ai-avatar-generator

接着创建一个Python虚拟环境，这能确保我们的项目依赖不会影响系统其他部分：

python3 -m venv venv

激活虚拟环境：

source venv/bin/activate

激活后，你的命令行提示符前面应该会出现(venv)字样，表示你现在在这个虚拟环境中工作。

4.2 安装必要的Python库

我们需要安装几个Python库：

gradio：创建网页界面
requests：发送HTTP请求
langchain：方便与大模型交互（可选，但推荐）

在虚拟环境中运行：

pip install gradio requests langchain-community

安装完成后，我们可以开始编写核心代码了。

4.3 编写头像生成器代码

创建一个名为avatar_generator.py的文件：

nano avatar_generator.py

如果你不熟悉nano编辑器，也可以使用vim avatar_generator.py或直接在其他编辑器中创建。

将以下代码复制到文件中：

import gradio as gr import requests import json import time # Ollama API地址 OLLAMA_URL = "http://localhost:11434/api/generate" # 头像风格模板 STYLE_TEMPLATES = { "赛博朋克": "cyberpunk style, neon lights, futuristic city background, digital aesthetic", "古风动漫": "ancient Chinese style anime, ink painting aesthetic, traditional clothing, serene landscape", "写实肖像": "photorealistic, professional portrait, studio lighting, detailed facial features", "卡通可爱": "cartoon style, cute and chibi, bright colors, simple background", "水墨风格": "Chinese ink wash painting, black and white, brush stroke texture, artistic", "科幻机甲": "sci-fi mecha, robotic elements, metallic texture, dynamic pose" } def generate_avatar_prompt(style_input, additional_details): """生成头像描述文案""" # 构建系统提示词 system_prompt = """你是一个专业的头像设计师和AI绘画提示词专家。 用户会描述他们想要的头像风格，你需要生成一份详细、专业、可直接用于Midjourney或Stable Diffusion的提示词。 请按照以下结构生成回答： 1. 人物特征描述（性别、年龄、发型、脸型、眼睛等） 2. 表情与姿态 3. 服装与配饰 4. 背景与环境 5. 光影与色彩 6. 艺术风格 7. 完整提示词（英文，适合直接复制到AI绘图工具） 确保描述具体、详细，包含丰富的视觉细节。""" # 构建用户输入 user_input = f"请为我设计一个{style_input}风格的头像。" if additional_details: user_input += f" 额外要求：{additional_details}" # 准备请求数据 payload = { "model": "qwen2.5:32b", "prompt": user_input, "system": system_prompt, "stream": False, "options": { "temperature": 0.7, "top_p": 0.9, "num_predict": 1024 } } try: # 发送请求到Ollama response = requests.post(OLLAMA_URL, json=payload, timeout=300) response.raise_for_status() result = response.json() return result.get("response", "生成失败，请重试。") except requests.exceptions.RequestException as e: return f"请求出错：{str(e)}" except json.JSONDecodeError: return "解析响应失败，请检查Ollama服务是否正常运行。" def quick_style_select(style_name): """快速选择预设风格""" if style_name in STYLE_TEMPLATES: return style_name, STYLE_TEMPLATES[style_name] return style_name, "" # 创建Gradio界面 with gr.Blocks(title="AI头像生成器", theme=gr.themes.Soft()) as demo: gr.Markdown("# 🎨 AI头像生成器") gr.Markdown("描述你想要的风格，AI生成详细的头像描述文案，可直接用于Midjourney、Stable Diffusion等AI绘图工具。") with gr.Row(): with gr.Column(scale=1): # 风格选择 style_type = gr.Dropdown( choices=list(STYLE_TEMPLATES.keys()), label="选择预设风格", value="赛博朋克" ) # 自定义风格输入 custom_style = gr.Textbox( label="或输入自定义风格描述", placeholder="例如：森林精灵风格，带有魔法光芒", lines=2 ) # 额外细节 additional_details = gr.Textbox( label="额外细节要求（可选）", placeholder="例如：蓝色头发，戴着眼镜，微笑表情", lines=3 ) # 快速填充按钮 gr.Markdown("### 快速填充") quick_buttons = gr.Row() for style in ["赛博朋克", "古风动漫", "写实肖像"]: quick_buttons.__enter__() btn = gr.Button(style, variant="secondary", size="sm") btn.click( fn=lambda s=style: (s, STYLE_TEMPLATES[s]), inputs=[], outputs=[style_type, custom_style] ) # 生成按钮 generate_btn = gr.Button("生成头像描述", variant="primary", size="lg") with gr.Column(scale=2): # 输出区域 output = gr.Textbox( label="生成的头像描述文案", placeholder="这里将显示生成的头像描述...", lines=20, max_lines=50 ) # 操作按钮 with gr.Row(): copy_btn = gr.Button("复制文案", variant="secondary") clear_btn = gr.Button("清空", variant="secondary") save_btn = gr.Button("保存为文件", variant="secondary") # 连接快速选择按钮 for style in STYLE_TEMPLATES: btn = gr.Button(style, visible=False) # 创建隐藏按钮用于连接 btn.click( fn=lambda s=style: s, inputs=[], outputs=[style_type] ) # 生成函数 def generate_full_prompt(style_type_val, custom_style_val, additional_val): """组合输入并生成提示词""" if custom_style_val: style_input = custom_style_val else: style_input = style_type_val return generate_avatar_prompt(style_input, additional_val) # 绑定事件 generate_btn.click( fn=generate_full_prompt, inputs=[style_type, custom_style, additional_details], outputs=[output] ) # 复制功能 copy_btn.click( fn=lambda x: x, inputs=[output], outputs=[], js="(x) => {navigator.clipboard.writeText(x); alert('文案已复制到剪贴板！');}" ) # 清空功能 clear_btn.click( fn=lambda: ("", "", ""), inputs=[], outputs=[custom_style, additional_details, output] ) # 保存功能 def save_to_file(text): if not text: return "没有内容可保存" filename = f"avatar_prompt_{int(time.time())}.txt" with open(filename, "w", encoding="utf-8") as f: f.write(text) return f"已保存为 {filename}" save_btn.click( fn=save_to_file, inputs=[output], outputs=[gr.Textbox(visible=False)] # 临时显示保存信息 ) # 示例 gr.Markdown("### 💡 使用示例") gr.Markdown(""" 1. **选择预设风格**：从下拉框选择“赛博朋克” 2. **添加细节**：在额外细节中输入“女性角色，紫色短发，机械义眼” 3. **点击生成**：等待AI生成详细描述 4. **复制使用**：将生成的英文提示词复制到Midjourney或Stable Diffusion """) # 启动应用 if __name__ == "__main__": demo.launch( server_name="0.0.0.0", server_port=8080, share=False )

按Ctrl+O保存文件，然后按Ctrl+X退出nano编辑器。

这段代码创建了一个完整的网页应用，主要功能包括：

选择预设的头像风格（赛博朋克、古风动漫等）
输入自定义风格描述
添加额外的细节要求
调用Qwen3-32B模型生成详细的头像描述
提供复制、清空、保存等实用功能

5. 第四步：运行和测试应用

代码写好了，现在让我们启动应用，看看效果如何。

5.1 启动头像生成器

首先确保Ollama服务正在运行（如果之前关闭了，需要重新启动）：

ollama serve

在另一个终端窗口中，进入项目目录并激活虚拟环境：

cd ~/ai-avatar-generator source venv/bin/activate

然后运行我们的应用：

python avatar_generator.py

你会看到类似这样的输出：

Running on local URL: http://0.0.0.0:8080

这表示应用已经成功启动，正在监听8080端口。

5.2 在浏览器中访问应用

现在打开你的Windows浏览器，输入以下地址：

http://localhost:8080

如果一切正常，你会看到一个美观的网页界面，标题是“AI头像生成器”。

让我们测试一下基本功能：

选择预设风格：在下拉框中选择“古风动漫”
添加细节：在“额外细节要求”中输入“女性角色，粉色汉服，手持团扇”
点击生成：点击“生成头像描述”按钮

等待几秒钟到一分钟（第一次生成可能需要更长时间，因为要加载模型），你会看到AI生成的详细头像描述。内容应该包括人物特征、表情姿态、服装配饰、背景环境、光影色彩等详细描述，最后还会给出完整的英文提示词。

5.3 使用生成的提示词

复制最后一部分的英文提示词，然后你可以：

在Midjourney的Discord频道中使用/imagine命令，粘贴提示词
在Stable Diffusion WebUI的提示词框中粘贴使用
或其他支持文本提示的AI绘图工具

这样，你就完成了从描述想法到生成可用的AI绘图提示词的完整流程。

6. 常见问题与优化建议

在实际使用中，你可能会遇到一些问题。这里我整理了一些常见情况和解决方法。

6.1 可能遇到的问题及解决

问题1：访问http://localhost:8080显示无法连接

解决方法：

检查应用是否成功启动（终端应该显示Running on local URL）
检查防火墙设置，确保8080端口没有被阻止
尝试使用http://127.0.0.1:8080

问题2：生成描述时提示“请求出错”或超时

解决方法：

确认Ollama服务正在运行：sudo systemctl status ollama
重启Ollama服务：sudo systemctl restart ollama
检查模型是否已下载：ollama list
增加超时时间：在代码中修改timeout参数（当前是300秒）

问题3：生成的内容不够详细或不符合预期

解决方法：

在“额外细节要求”中提供更具体的描述
尝试不同的风格组合
修改代码中的system_prompt，让AI更清楚你的需求

问题4：内存不足导致运行缓慢或崩溃

解决方法：

Qwen3-32B需要较大内存，确保至少有16GB可用内存
关闭其他占用内存大的程序
如果内存不足，可以考虑使用较小的模型版本，如Qwen3-14B

6.2 性能优化建议

如果你觉得生成速度较慢，可以尝试以下优化：

调整生成参数：在代码中找到payload部分，调整这些参数：

"num_predict": 512：减少生成的最大token数（当前是1024）
"temperature": 0.8：提高温度值，增加创造性（当前是0.7）
"top_p": 0.95：提高top_p值，增加多样性（当前是0.9）

使用GPU加速：如果你有NVIDIA显卡，可以配置Ollama使用GPU：

安装NVIDIA驱动和CUDA
停止Ollama服务：sudo systemctl stop ollama
编辑配置：sudo nano /etc/systemd/system/ollama.service
在[Service]部分添加：Environment="OLLAMA_HOST=0.0.0.0"和Environment="OLLAMA_GPU_LAYERS=99"
重启服务：sudo systemctl daemon-reload && sudo systemctl start ollama

添加更多预设风格：你可以扩展STYLE_TEMPLATES字典，添加更多风格选项：

STYLE_TEMPLATES = { # 原有风格... "蒸汽朋克": "steampunk style, brass gears, Victorian era, mechanical elements", "奇幻魔法": "fantasy magic style, glowing runes, mystical atmosphere, wizard robes", "极简主义": "minimalist style, simple lines, solid colors, clean background", "水彩画风": "watercolor painting style, soft edges, blended colors, artistic", # 添加更多你喜欢的风格... }