企业官网建设流程全解析

Qwen2.5-1.5B部署教程：WSL2 GPU加速启用+nvtop实时显存监控配置

1. 项目概述

Qwen2.5-1.5B是阿里通义千问推出的轻量级大语言模型，特别适合在本地环境中部署运行。本教程将指导您完成在WSL2环境下启用GPU加速，并配置nvtop实时显存监控工具，打造一个高效的本地智能对话系统。

这个方案有三大核心优势：

• 完全本地运行：所有数据处理和模型推理都在您的设备上完成
• 低资源消耗：1.5B参数规模对显存要求较低
• 易用界面：基于Streamlit的友好交互界面

2. 环境准备

2.1 硬件要求

建议配置：

• NVIDIA显卡（GTX 1060 6GB或更高）
• 16GB以上系统内存
• 至少10GB可用磁盘空间

2.2 软件依赖

确保已安装：

• Windows 10/11 with WSL2
• Ubuntu 20.04/22.04 on WSL2
• NVIDIA显卡驱动（Windows端）
• CUDA Toolkit 11.7或更高版本

3. WSL2 GPU环境配置

3.1 启用WSL2 GPU支持

1. 以管理员身份打开PowerShell，执行：

wsl --update wsl --shutdown

2. 创建或修改%USERPROFILE%\.wslconfig文件：

[wsl2] memory=8GB swap=4GB localhostForwarding=true

3. 在Ubuntu中安装CUDA工具链：

sudo apt update sudo apt install -y nvidia-cuda-toolkit

3.2 验证GPU可用性

运行以下命令检查GPU识别情况：

nvidia-smi

预期输出应显示您的GPU型号和驱动版本。

4. 模型部署

4.1 下载模型文件

从阿里云官方渠道获取Qwen2.5-1.5B-Instruct模型，建议使用git-lfs：

sudo apt install -y git-lfs git lfs install git clone https://huggingface.co/Qwen/Qwen2.5-1.5B-Instruct /root/qwen1.5b

4.2 安装Python依赖

创建并激活Python虚拟环境：

python -m venv qwen_env source qwen_env/bin/activate

安装必要包：

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 pip install transformers streamlit nvidia-ml-py3

5. 配置实时显存监控

5.1 安装nvtop

nvtop是类似htop的GPU监控工具：

sudo apt install -y cmake libncurses5-dev libncursesw5-dev git clone https://github.com/Syllo/nvtop.git mkdir -p nvtop/build && cd nvtop/build cmake .. -DNVML_RETRIEVE_HEADER_ONLINE=True make sudo make install

5.2 使用nvtop监控

在新终端中运行：

nvtop

这将显示GPU使用率、显存占用、温度等实时数据。

6. 启动对话服务

6.1 创建启动脚本

新建qwen_app.py文件：

import torch from transformers import AutoModelForCausalLM, AutoTokenizer import streamlit as st MODEL_PATH = "/root/qwen1.5b" @st.cache_resource def load_model(): print(f" 正在加载模型: {MODEL_PATH}") tokenizer = AutoTokenizer.from_pretrained(MODEL_PATH, trust_remote_code=True) model = AutoModelForCausalLM.from_pretrained( MODEL_PATH, device_map="auto", torch_dtype="auto", trust_remote_code=True ) return model, tokenizer model, tokenizer = load_model() st.title("Qwen2.5-1.5B 本地对话助手") if "messages" not in st.session_state: st.session_state.messages = [] for message in st.session_state.messages: with st.chat_message(message["role"]): st.markdown(message["content"]) if prompt := st.chat_input("你好，我是Qwen..."): st.session_state.messages.append({"role": "user", "content": prompt}) with st.chat_message("user"): st.markdown(prompt) with st.chat_message("assistant"): with torch.no_grad(): inputs = tokenizer.apply_chat_template( st.session_state.messages, add_generation_prompt=True, return_tensors="pt" ).to(model.device) outputs = model.generate( inputs, max_new_tokens=1024, temperature=0.7, top_p=0.9 ) response = tokenizer.decode(outputs[0][len(inputs[0]):], skip_special_tokens=True) st.markdown(response) st.session_state.messages.append({"role": "assistant", "content": response}) with st.sidebar: if st.button("🧹 清空对话"): st.session_state.messages = [] torch.cuda.empty_cache()

6.2 启动服务

运行以下命令启动Web界面：

streamlit run qwen_app.py

首次启动需要加载模型，请耐心等待。之后可以通过浏览器访问本地URL使用对话服务。

7. 常见问题解决

7.1 CUDA内存不足

如果遇到CUDA out of memory错误：

1. 减少max_new_tokens参数值
2. 点击侧边栏"清空对话"按钮
3. 重启服务释放显存

7.2 模型加载缓慢

首次加载较慢是正常现象，后续启动会利用缓存加速。如果持续缓慢：

sudo apt install -y nvidia-persistenced sudo systemctl enable nvidia-persistenced

7.3 WSL2网络问题

如果无法访问服务：

sudo apt install -y net-tools ifconfig | grep inet

检查IP地址是否正确。

8. 总结

通过本教程，您已经成功在WSL2环境中：

1. 配置了GPU加速支持
2. 部署了Qwen2.5-1.5B本地模型
3. 设置了实时显存监控
4. 启动了基于Streamlit的对话界面

这套方案特别适合需要本地私有化部署AI助手的场景，兼顾了性能与隐私保护。您可以根据实际需求调整模型参数或界面样式，打造专属的智能对话体验。

获取更多AI镜像

想探索更多AI镜像和应用场景？访问 CSDN星图镜像广场，提供丰富的预置镜像，覆盖大模型推理、图像生成、视频生成、模型微调等多个领域，支持一键部署。