VectorDBBench 测试环境搭建
压测工具:VectorDBBench:
操作系统:Ubuntu 20.04
规格:16C32G(压测主机也需要高配置,否则内存打满会宕机)
sudo apt install -y software-properties-common
sudo add-apt-repository ppa:deadsnakes/ppa
sudo apt update
apt install python3.12 python3.12-dev
python3.12 -m pip install gencode -i https://pypi.tuna.tsinghua.edu.cn/simple
python3.12 -m pip install pymilvus==2.6.0b0 -i https://pypi.tuna.tsinghua.edu.cn/simple
或 pip install pymilvus==2.6.0b0 -i https://pypi.tuna.tsinghua.edu.cn/simple
((312) ) root@milvus-test-wushan:~/VectorDBBench# pip3 list
Package Version
————— ———–
click 8.2.1
cx_Oracle 8.3.0
db-hammer 0.0.60
genCode 1.0.1
grpcio 1.67.1
Jinja2 3.1.6
MarkupSafe 3.0.2
milvus-lite 2.5.1
numpy 2.3.2
pandas 2.3.1
pip 25.2
protobuf 6.31.1
pymilvus 2.6.0b0
PyMySQL 1.1.1
python-dateutil 2.9.0.post0
python-dotenv 1.1.1
pytz 2025.2
PyYAML 6.0.2
setuptools 80.9.0
six 1.17.0
tqdm 4.67.1
tzdata 2025.2
ujson 5.10.0
((312) ) root@milvus-test-wushan:~/VectorDBBench#
===安装conda
mkdir -p ~/miniconda3
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O ~/miniconda3/miniconda.sh
bash ~/miniconda3/miniconda.sh -b -u -p ~/miniconda3 rm -rf ~/miniconda3/miniconda.sh
After installing, initialize your newly-installed Miniconda. The following commands initialize for bash : ~/miniconda3/bin/conda init bash
conda create --name VDBBench python=3.12 # 激活测试环境 conda activate VDBBench # 验证 Python 版本 python --version
环境准备完成后,安装 VectorDBBench。建议使用国内镜像源以提高下载速度。
pip install vectordb-bench -i https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple
安装完成后,VectorDBBench 可以通过命令行方式运行测试。为了更直观地管理测试任务和查看结果,VectorDBBench 还提供了基于 Web 的可视化界面——Leaderboard。
部署 Leaderboard:Web可视化界面
Leaderboard 提供了友好的 Web 界面,可以通过图形化方式配置测试参数、监控测试进度、查看测试结果。相比于命令行方式,这种可视化操作更加直观和便捷。
# 安装测试相关依赖(非必须) pip install -e '.[test]' -i https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple pip install -e '.[pinecone]' -i https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple
# 启动 Leaderboard Web 服务 python -m vectordb_bench
访问公网IP的对应端口,进行压测。