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的对应端口,进行压测。

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注