安装支持 GPU 的 Milvus

系统要求

硬件要求

组件 建议配置
CPU Intel CPU Haswell 或以上
GPU NVIDIA Pascal 或以上
内存 8 GB 或以上 （取决于具体向量数据规模）
硬盘 SATA 3.0 SSD 或以上

Milvus Docker 要求

  • 在您的宿主机上 安装 Docker 19.03 或更高版本。
  • 安装 NVIDIA driver 418 或更高版本。
  • 。

第一步 确认 Docker 状态

如果无法正常打印 Docker 相关信息，请启动 Docker daemon.

拉取支持 GPU 的镜像：

  1. $ docker pull milvusdb/milvus:0.7.0-gpu-d031120-40c7e8

第三步 下载并修改配置文件

您可以使用以下方法下载配置文件：

配置文件下载完成后，您需要将 server_config.yaml 中的 gpu_resource_config 部分的 参数设置为 true。

  1. $ docker run -d --name milvus_gpu --gpus all \
  2. -p 19530:19530 \
  3. -p 19121:19121 \
  4. -p 9091:9091 \
  5. -v /home/$USER/milvus/db:/var/lib/milvus/db \
  6. -v /home/$USER/milvus/conf:/var/lib/milvus/conf \
  7. -v /home/$USER/milvus/logs:/var/lib/milvus/logs \
  9. milvusdb/milvus:0.7.0-gpu-d031120-40c7e8
  • -d: 运行 container 到后台并打印 container id。
  • --name: 为 container 分配一个名字。
  • --gpus: 指定可用的 GPU。如未填写任何值，则所有 GPU 都可用。
  • -p: 暴露 container 端口到 host。

最后，确认 Milvus 运行状态：

如果 Milvus 服务没有正常启动，您可以执行以下命令查询错误日志。

  1. # Get id of the container running Milvus
  2. $ docker ps -a
  4. $ docker logs <milvus container id>

接下来您可以