yusefallgood/pgvector

pgvector 0.5.1 on Postgres 16 (Aurora V2兼容) 自定义镜像

镜像概述

本镜像为基于PostgreSQL 16构建的自定义Docker镜像，预集成pgvector 0.5.1扩展，专为与Amazon Aurora V2兼容设计，提供向量数据存储与查询能力，适用于需要在Aurora V2环境中部署向量数据库功能的场景。

核心功能和特性

• 基础环境：基于PostgreSQL 16官方镜像构建，包含PostgreSQL全部核心功能
• 向量扩展：预安装pgvector 0.5.1扩展，支持向量数据类型（vector）及向量操作（如欧氏距离、余弦相似度计算等）
• 兼容性：针对Aurora V2环境优化，确保与AWS Aurora PostgreSQL兼容
• 标准兼容：支持PostgreSQL标准客户端连接协议，可无缝集成现有PostgreSQL工具链

使用场景和适用范围

• AI/机器学习应用中的向量数据存储（如embedding向量）
• 相似性搜索场景（如图片、文本相似度匹配）
• 需要在Aurora V2环境中使用pgvector 0.5.1特定版本的业务系统
• 对PostgreSQL 16特性有依赖且需向量数据库功能的应用

使用方法和配置说明

基本使用（Docker Run）

bash
docker run -d \
  --name pgvector-aurora \
  -p 5432:5432 \
  -e POSTGRES_USER=myuser \
  -e POSTGRES_PASSWORD=mypassword \
  -e POSTGRES_DB=mydb \
  -v pgdata:/var/lib/postgresql/data \
  [镜像名称]

环境变量配置

启用pgvector扩展

连接数据库后执行以下SQL启用扩展：

sql
CREATE EXTENSION vector;

Docker Compose配置示例

yaml
version: '3.8'
services:
  pgvector:
    image: [镜像名称]
    container_name: pgvector-aurora
    ports:
      - "5432:5432"
    environment:
      POSTGRES_USER: myuser
      POSTGRES_PASSWORD: mypassword
      POSTGRES_DB: mydb
    volumes:
      - pgdata:/var/lib/postgresql/data
    restart: unless-stopped

volumes:
  pgdata:

验证安装

连接数据库后验证pgvector版本：

sql
SELECT * FROM pg_extension WHERE extname = 'vector';

应返回版本为0.5.1的扩展记录。