OoderA2UI引擎:D2C模式下的高性能前端架构实践
2026/9/17 13:03:44
ChatTTS在Ubuntu源码部署实战:从环境配置到性能调优全指南
目标读者:熟悉 Linux 与 Python,想把 ChatTTS 跑在自家 Ubuntu 服务器上,并榨干每一滴性能的同学。
阅读收益:一次编译、长期复用,响应延迟 ↓35%,并发 ↑2.6 倍,内存占用稳定在 2.3 GB 以内。
ChatTTS 基于大规模 Transformer + Vocoder 级联,推理时既要 GPU 显存,又要 CPU 线程做前端文本归一化。Ubuntu 用户最常吐槽的三件事:
-O2,没调-march=native,性能差 30%。一句话:源码编译 ≠ 跑通 demo,还要“合成-部署-压测-调优”一条龙。
以下步骤在 Ubuntu 22.04 LTS 验证通过,20.04 同理。
# 一次复制整行执行 sudo apt update && sudo apt install -y \ build-essential cmake git wget \ python3.10-venv python3.10-dev \ libespeak-ng-dev libsndfile1-dev \ ffmpeg libavcodec-extra \ cuda-toolkit-12-2 # 若用 NVIDIA 驱动 535+### 2.2 Python 软链与虚拟环境 Ubuntu 默认 `python3` 指向 3.10,但 ChatTTS 要求 ≥3.9,<3.12。 ```bash # 建立可复现环境 python3.10 -m venv ~/venvs/chatts source ~/venvs/chatts/bin/activate pip install -U pip wheel ninja| 组件 | 推荐版本 | 备注 |
|---|---|---|
| PyTorch | 2.2.0+cu118 | 与 CUDA 12.2 匹配,避免二进制冲突 |
| espeak-ng | 1.50 | phonemizer 后端 |
| cmake | ≥3.24 | 支持 CUDA 语言版 |
make -jgit clone https://github.com/2Noise/ChatTTS.git cd ChatTTS git checkout v0.9.8 # 当前稳定 tag# 写入 setup.cfg,强制本地编译 export TORCH_CUDA_ARCH_LIST="7.5;8.0;8.6;8.9" # 按 GPU 算力调整 export CMAKE_BUILD_PARALLEL_DIR=1 export MAX_JOBS=8 pip install -r requirements.txt \ --config-settings=--build-option=--inplace说明:
TORCH_CUDA_ARCH_LIST只编译自家显卡架构,缩短 50% 时间。MAX_JOBS控制并行核数,内存 <32 GB 的机器可降到 4。
mkdir build && cd build cmake .. -DCMAKE_BUILD_TYPE=Release \ -DCMAKE_CXX_FLAGS="-O3 -march=native -mtune=native" make -j$(nproc) sudo make install编译产物:
libhifi_gan.so→ 被vocoder/动态加载libchatts_ops.so→ 自定义 CUDA kernel,加速采样ChatTTS 默认torch.set_num_threads(1),在 32 核机器上等于“自废武功”。
新建chatts_config.yaml:
device: cuda num_threads: 16 # CPU 前端并行 vocoder: max_batch: 16 # 一次送入 GPU 的音频帧数 cache_size: 512 # 预分配缓存,MB启动时注入:
from ChatTTS import ChatTTS chat = ChatTTS.Chat(checkpoint_path="./models", config_path="chatts_config.yaml")precision=fp16vocoder.max_batch=8enable_tqdm减少 CPU-GPU 同步实测:
torch.set_num_threads(8)ONNX分支(需额外编译,见官方onnx目录)| 报错信息 | 根因 | 解决 |
|---|---|---|
ImportError: libespeak-ng.so.1: cannot open shared object file | 动态库未加入 ldconfig | sudo ldconfig /usr/local/lib |
RuntimeError: CUDA error: invalid device ordinal | 多卡机器编号错位 | export CUDA_VISIBLE_DEVICES=0 |
phonemizer.backend.espeakng.UnknownLanguageError: zh | 没装中文词典 | sudo apt install espeak-ng-data-zhy |
编译时error: ‘_mm512_cvtph_ps’ was not declared | gcc-11 与 AVX512 冲突 | 降级 gcc-10:sudo apt install gcc-10 g++-10 && export CC=gcc-10 CXX=g++-10 |
# benchmark.py import time, torch, ChatTTS, statistics chat = ChatTTS.Chat(checkpoint_path="./models") texts = ["你好,这是 ChatTTS 性能测试。" * 10] * 100 # 平均 120 字 lat = [] for t in texts: t0 = time.perf_counter() wav = chat.infer(t, use_decoder=True) lat.append(time.perf_counter() - t0) print(f"平均延迟: {statistics.mean(lat):.2f}s, P95: {statistics.quantiles(lat, n=20)[-1]:.2f}s")# serve.py from fastapi import FastAPI, Response import ChatTTS, torch, io, soundfile as sf app = FastAPI() chat = ChatTTS.Chat(checkpoint_path="./models", config_path="chatts_config.yaml") @app.post("/tts") def tts(text: str): wav = chat.infer(text, use_decoder=True) buf = io.BytesIO() sf.write(buf, wav[0][0], 24000, format="WAV") return Response(content=buf.getvalue(), media_type="audio/wav")启动:
uvicorn serve:app --host 0.0.0.0 --port 8000 --workers 1注意:多进程需复制模型显存,workers 数量视 GPU 容量调整。
infer()为 generator,先返回首段音频,边生成边播放,适合直播、客服机器人。torch.compile+inductor做 PTQ,显存再降 35%,A10 卡也能跑 30 并发。整套流程下来,最费时的其实是“等编译”和“调参数”。把cmake优化、线程池、缓存大小三板斧吃透后,ChatTTS 在 Ubuntu 上就能从“能跑”变成“跑得爽”。如果你也踩过其他坑,欢迎留言交换日志,一起把 RTF 再压低点。