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

Wan2.1-UMT5入门指南：Ubuntu 20.04系统下的Docker环境配置与部署

news 2026/4/14 7:56:47

Wan2.1-UMT5入门指南：Ubuntu 20.04系统下的Docker环境配置与部署

最近在折腾一些多语言翻译相关的AI应用，发现Wan2.1-UMT5这个模型挺有意思的，它在处理一些专业术语和长文本翻译上表现不错。不过，对于很多刚接触的朋友来说，怎么把它在服务器上跑起来，特别是用Docker这种现在很流行的方式，可能还有点摸不着头脑。

今天这篇内容，我就以最常用的Ubuntu 20.04系统为例，带你走一遍完整的部署流程。你不用有太深的Docker知识，只要跟着步骤操作，就能在自己的服务器上把环境搭好，让模型跑起来。整个过程我会尽量讲得细一些，把可能遇到的坑也提前告诉你。

1. 准备工作：系统与Docker环境

在开始部署模型之前，我们得先把“地基”打好。这个地基就是Ubuntu 20.04系统和Docker环境。别担心，步骤都很简单，大部分就是复制粘贴命令。

1.1 确保你的Ubuntu系统是最新的

首先，我们登录到你的Ubuntu 20.04服务器。打开终端，输入的第一条命令永远是更新系统软件包列表。这就像你去超市前，先看看最新的商品目录一样。

sudo apt update

执行完上面那条，我们接着把系统里已有的软件都升级到最新版本。这个命令可能会花点时间，取决于你的网络速度和需要更新的软件数量。

sudo apt upgrade -y

那个-y参数的意思是，当系统问你是否确认安装时，自动回答“是”，这样就不用守在电脑前一直按回车了。

1.2 安装Docker引擎

系统更新好了，接下来安装Docker。Docker官方提供了非常方便的安装脚本，我们直接用它就行。

下载安装脚本：
```
curl -fsSL https://get.docker.com -o get-docker.sh
```
这条命令会把安装脚本下载到你的当前目录，文件名叫get-docker.sh。
运行安装脚本：
```
sudo sh get-docker.sh
```
脚本会自动完成所有依赖的检查和安装。安装完成后，我们需要做一件很重要的事：把当前用户添加到docker用户组。这样以后运行Docker命令就不用每次都加sudo了，会方便很多。
```
sudo usermod -aG docker $USER
```
注意，执行完这条命令后，你需要完全退出当前终端会话，然后重新登录，这个分组变更才会生效。你可以直接关闭终端再打开，或者断开SSH连接重新连一次。
验证安装：重新登录后，运行下面的命令检查Docker是否安装成功，并且可以正常使用了。
```
docker --version
```
如果看到类似Docker version 20.10.17, build 100c701的输出，说明安装没问题。再跑一下经典的hello-world镜像试试：
```
docker run hello-world
```
如果能看到一段欢迎信息，恭喜你，Docker环境已经准备就绪。

2. 配置NVIDIA容器运行时

如果你的服务器有NVIDIA显卡，并且希望模型能使用GPU来加速计算（这能极大提升翻译速度），那么这一步是必须的。如果服务器没有GPU，你可以跳过这一节，模型会使用CPU运行，只是速度会慢一些。

2.1 安装NVIDIA驱动和CUDA工具包

首先，确保你的服务器已经安装了合适的NVIDIA显卡驱动。你可以用这个命令检查：

nvidia-smi

如果显示了显卡信息和驱动版本，那就可以直接进行下一步。如果没有输出，你需要先安装驱动。对于Ubuntu 20.04，一个比较简单的方法是使用系统自带的附加驱动工具，或者参考NVIDIA官网的指南。

接下来，我们需要安装NVIDIA Container Toolkit。它能让Docker容器识别并使用宿主机的GPU。

添加软件仓库和密钥：

distribution=$(. /etc/os-release;echo $ID$VERSION_ID) curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add - curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list

安装工具包：

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

2.2 配置Docker使用NVIDIA运行时

安装好工具包后，需要告诉Docker默认使用NVIDIA的容器运行时。

配置Docker守护进程：
```
sudo nvidia-ctk runtime configure --runtime=docker sudo systemctl restart docker
```
第一条命令会修改Docker的配置文件，第二条命令是重启Docker服务让配置生效。
测试GPU在容器中是否可用：运行一个测试命令，看看容器里能不能看到GPU。
```
docker run --rm --gpus all nvidia/cuda:11.8.0-base-ubuntu20.04 nvidia-smi
```
这个命令会启动一个带有CUDA基础环境的小容器，并在里面执行nvidia-smi。如果输出的显卡信息和你直接在宿主机上运行nvidia-smi的结果一致，那就说明配置成功了，Docker容器现在可以愉快地使用你的显卡了。

3. 获取并运行Wan2.1-UMT5镜像

基础环境全部搞定，现在终于到了主角登场环节。我们将从镜像仓库拉取预先打包好的Wan2.1-UMT5 Docker镜像，并把它运行起来。

3.1 拉取专用镜像

这里我们使用一个已经集成好所有依赖的专用镜像。拉取镜像的命令很简单，就像下载一个软件安装包。

docker pull csdnmirrors/wan2.1-umt5:latest

docker pull是拉取命令，后面的csdnmirrors/wan2.1-umt5:latest是镜像的地址和标签。latest标签代表最新版本。执行这条命令后，Docker会开始从网络下载镜像，你需要等待一段时间，镜像大小有几个GB。

3.2 启动模型服务容器

镜像下载完成后，我们用一个命令来启动它。这个命令看起来有点长，但每一部分都有它的作用，我拆开给你讲。

docker run -d \ --name wan-umt5 \ --gpus all \ -p 7860:7860 \ -v /home/your_username/wan_model_data:/app/data \ csdnmirrors/wan2.1-umt5:latest

我们来解释一下每个参数：

-d：让容器在“后台”运行，这样你关闭终端后服务也不会停。
--name wan-umt5：给这个容器起个名字，方便以后管理，比如停止、重启它。
--gpus all：把宿主机的所有GPU都分配给这个容器使用。如果你没有GPU或者想用CPU，去掉这个参数即可。
-p 7860:7860：这是“端口映射”。容器内部的服务运行在7860端口上，我们把它映射到宿主机的7860端口。这样，你通过浏览器访问服务器IP:7860就能看到模型的Web界面了。
-v /home/your_username/wan_model_data:/app/data：这是“卷映射”，非常重要。它把宿主机上的一个目录（比如/home/your_username/wan_model_data）挂载到容器内的/app/data目录。模型生成的数据、缓存或者你的自定义配置都会保存在这里，即使容器被删除，你的数据也不会丢失。请把/home/your_username/替换成你服务器上真实的用户名目录。
最后一行csdnmirrors/wan2.1-umt5:latest就是指定我们刚刚拉取的镜像。

命令执行后，如果没有报错，会输出一串容器的ID。你可以用下面这个命令查看容器是否在正常运行：

docker ps

你应该能看到一个名为wan-umt5的容器，状态是Up。

4. 访问与验证服务

容器跑起来了，我们怎么知道它真的在工作呢？很简单，打开浏览器访问一下。

4.1 通过Web界面访问

在你的电脑浏览器地址栏输入：http://你的服务器IP地址:7860。

你的服务器IP地址：就是你Ubuntu服务器的公网IP或者内网IP。
:7860：就是我们刚才映射的端口。

如果一切顺利，稍等片刻（第一次启动可能需要加载模型，时间稍长），你就能看到一个Web用户界面。这个界面通常就是Wan2.1-UMT5模型的交互界面，你可以在里面输入文本进行翻译测试。

4.2 进行简单的功能测试

进入Web界面后，你可以找一个简短的句子试试看。比如输入一句中文“你好，世界！”，选择翻译成英文，看看输出是不是“Hello, World!”。

如果页面能打开，并且翻译功能有响应，哪怕第一次因为模型加载慢一点，也基本说明服务部署成功了。如果无法访问页面，可以回到服务器终端，用下面的命令查看容器的日志，里面通常会有错误信息提示。

docker logs wan-umt5

5. 生产环境实用配置

把服务跑起来只是第一步，要让它在服务器上稳定、可靠地长期运行，我们还需要做一些“后勤保障”工作。

5.1 设置容器自动重启

服务器有时候需要重启，我们希望容器能随着服务器一起自动启动。Docker提供了重启策略。

我们先停止并删除刚才启动的容器（别担心，数据在映射的卷里，不会丢）：

docker stop wan-umt5 docker rm wan-umt5

然后用一个增强版的命令重新启动它，主要增加了--restart unless-stopped参数：

docker run -d \ --name wan-umt5 \ --restart unless-stopped \ --gpus all \ -p 7860:7860 \ -v /home/your_username/wan_model_data:/app/data \ csdnmirrors/wan2.1-umt5:latest

--restart unless-stopped这个策略的意思是：除非你手动停止这个容器，否则如果容器自己退出了（比如进程崩溃），或者宿主机重启了，Docker都会自动重新启动它。这能保证服务的可用性。