服务器上部署GPU版的milvus向量数据库

1、安装docker compose

• 我们可以从 Github 上下载它的二进制包来利用，最新发行的版本地点：
  https://github.com/docker/compose/releases
  

1. sudo curl -L "https://github.com/docker/compose/releases/download/v2.6.0/docker-compose-$(uname -s)-$(uname -m)" -o /usr/local/bin/docker-compose

复制代码

• 将可执行权限应用于二进制文件：
  

1. $ sudo chmod +x /usr/local/bin/docker-compose

复制代码

• 创建软链：
  

1. $ sudo ln -s /usr/local/bin/docker-compose /usr/bin/docker-compose

复制代码

• 测试是否安装成功：
  

1. $ docker-compose version

复制代码

2、安装GPU驱动

已安装好的朋友，这步可以跳过

1. sudo apt install --no-install-recommends nvidia-headless-545 nvidia-utils-545

复制代码

3、下载安装milvus

1.Download milvus-standalone-docker-compose-gpu.yml

1. wget https://github.com/milvus-io/milvus/releases/download/v2.4.0-rc.1/milvus-standalone-docker-compose-gpu.yml -O docker-compose.yml

复制代码

各人根据自己的硬件设置，利用单GPU，还是利用多GPU都可以。在docker-compose.yml文件中更改。
单GPU：

1. ...
2. standalone:
3.   gpu:
4.     initMemSize: 0
5.     maxMemSize: 1024
6.   ...
7.   deploy:
8.     resources:
9.       reservations:
10.         devices:
11.           - driver: nvidia
12.             capabilities: ["gpu"]
13.             device_ids: ["0"]
14. ...

复制代码

多GPU：

1. ...
2. standalone:
3.   gpu:
4.     initMemSize: 0
5.     maxMemSize: 1024
6.   ...
7.   deploy:
8.     resources:
9.       reservations:
10.         devices:
11.           - driver: nvidia
12.             capabilities: ["gpu"]
13.             device_ids: ['0', '1']
14. ...

复制代码

2.Start Milvus

在docker-compose.yml路径下，

1. sudo docker compose up -d

复制代码

或

1. sudo docker-compose up -d

复制代码

启动后会表现：
   Creating milvus-etcd … done
Creating milvus-minio … done
Creating milvus-standalone … done
  3.验证安装

Milvus 单机版启动后，将有三个 docker 容器在运行，包括 Milvus 单机版服务及其两个依赖项。

1. sudo docker ps

复制代码

1.       Name                     Command                  State                            Ports
2. --------------------------------------------------------------------------------------------------------------------
3. milvus-etcd         etcd -advertise-client-url ...   Up             2379/tcp, 2380/tcp
4. milvus-minio        /usr/bin/docker-entrypoint ...   Up (healthy)   9000/tcp
5. milvus-standalone   /tini -- milvus run standalone   Up             0.0.0.0:19530->19530/tcp, 0.0.0.0:9091->9091/tcp

复制代码

4.Stop Milvus

关闭Milvus standalone：

1. sudo docker compose down

复制代码

在关闭Milvus后删除数据：

1. sudo rm -rf  volumes

复制代码

---

参考链接：https://milvus.io/docs/install_standalone-docker-compose-gpu.md

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。