安装 GPU 版本的 Milvus

安装前提

操作系统

操作系统版本
CentOS7.5 或以上
Ubuntu LTS18.04 或以上

硬件

硬件建议配置
CPUIntel CPU Sandy Bridge 或以上
CPU 指令集
  • SSE42
  • AVX
  • AVX2
  • AVX512
GPUNVIDIA Pascal 或以上
内存8 GB 或以上(取决于具体向量数据规模)
硬盘SATA 3.0 SSD 或以上

软件

软件版本
Docker19.03 或以上
NVIDIA driver418 或以上

确认 Docker 状态

确认 Docker daemon 正在运行:

  1. $ sudo docker info
  • 如果无法正常打印 Docker 相关信息,请启动 Docker daemon。
  • 在 Linux 上需要使用 sudo 执行 Docker 命令。若要在没有 sudo 的情况下运行 Docker 命令,请创建 docker 组并添加用户,详见 Linux 安装步骤

拉取 Milvus 镜像

拉取支持 GPU 的镜像:

  1. $ sudo docker pull milvusdb/milvus:0.10.0-gpu-d061620-5f3c00

如果拉取镜像的速度过慢或一直失败,请参考 操作常见问题 中提供的解决办法。

下载并修改配置文件

  1. $ mkdir -p /home/$USER/milvus/conf
  2. $ cd /home/$USER/milvus/conf
  3. $ wget http://raw.githubusercontent.com/milvus-io/milvus/v0.10.0/core/conf/demo/server_config.yaml

如果无法通过 wget 命令正常下载,你也可以在 /home/$USER/milvus/conf 目录下创建 server_config.yaml 文件,然后将 server config 文件 的内容复制到你创建的配置文件中。

配置文件下载完成后,你需要将 server_config.yaml 中的 gpu 区域的 enable 参数设置为 true

启动 Milvus Docker 容器

启动 Docker 容器之前,你必须将 server_config.yamlgpu 部分的 enable 设为 true

启动 Docker 容器,将本地的文件路径映射到容器中:

  1. $ sudo docker run -d --name milvus_gpu_0.10.0 --gpus all \
  2. -p 19530:19530 \
  3. -p 19121:19121 \
  4. -v /home/$USER/milvus/db:/var/lib/milvus/db \
  5. -v /home/$USER/milvus/conf:/var/lib/milvus/conf \
  6. -v /home/$USER/milvus/logs:/var/lib/milvus/logs \
  7. -v /home/$USER/milvus/wal:/var/lib/milvus/wal \
  8. milvusdb/milvus:0.10.0-gpu-d061620-5f3c00

上述命令中用到的参数定义如下:

  • -d: 在后台运行 container。
  • --name: 为 container 指定一个名字。
  • --gpus: 指定可用的 GPU。如未填写任何值,则所有 GPU 都可用。
  • -p: 映射宿主机端口到 container。
  • -v: 将宿主机路径挂载至 container。

最后,确认 Milvus 运行状态:

  1. $ sudo docker ps

如果 Milvus 服务没有正常启动,执行以下命令查询错误日志:

  1. $ sudo docker logs milvus_gpu_0.10.0

接下来你可以