安装 Milvus

浏览: 1122

https://milvus.io/docs/v0.10.2/milvus_docker-gpu.md

确认后台是否运行了Docker daemon

sudo docker info
sudo systemctl start docker # 启动docker
sudo docker ps -a # 查看所有的容器

获取支持GPU的milvus 镜像

sudo docker pull milvusdb/milvus:0.10.2-gpu-d081520-8a2393

安装milvus

mkdir -p /home/$USER/milvus/conf
cd /home/$USER/milvus/conf
wget https://raw.githubusercontent.com/milvus-io/milvus/0.10.2/core/conf/demo/server_config.yaml

修改设置,使得gpu可用:

vi /home/$USER/milvus/conf/server_config.yaml
gpu:
enable: true
:wq

查看docker服务状态

sudo service docker status

重启docker服务

sudo service docker restart

启动Docker容器

sudo docker run -d --name milvus_gpu_0.10.2 --gpus all \
-p 19530:19530 \
-p 19121:19121 \
-v /home/$USER/milvus/db:/var/lib/milvus/db \
-v /home/$USER/milvus/conf:/var/lib/milvus/conf \
-v /home/$USER/milvus/logs:/var/lib/milvus/logs \
-v /home/$USER/milvus/wal:/var/lib/milvus/wal \
milvusdb/milvus:0.10.2-gpu-d081520-8a2393

参数:

  • -d: Runs container in the background and prints container ID.
  • --name: Assigns a name to the container.
  • --gpus: Assigns GPU devices to the container. (all represents all GPUs.)
  • -p: Publishes a container’s port(s) to the host.
  • -v: Mounts the directory into the container.
milvusdb/milvus:0.10.2-gpu-d081520-8a2393    # image_id

查看容器状态:

sudo docker stats milvus_gpu_0.10.2

查看进程信息:

sudo docker top milvus_gpu_0.10.2

列出已经下载下来的镜像:https://yeasy.gitbook.io/docker_practice/image/list

sudo docker image ls -a

查看镜像、容器、数据卷所占用的空间

sudo docker system df

显示、删除悬虚镜像

sudo docker image ls -f dangling=true

sudo docker image prune

重启电脑后:

解决 nvidia-smi 报错:Failed to initialize NVML: Driver/library version mismatch.

sudo rmmod nvidia_drm
sudo rmmod nvidia_uvm
sudo rmmod nvidia_modeset
sudo rmmod nvidia

启动docker

sudo service docker restart

删除已有的milvus容器

sudo docker rm milvus_gpu_0.10.2

启动容器

sudo docker run -d --name milvus_gpu_0.10.2 --gpus all \
-p 19530:19530 \
-p 19121:19121 \
-v /home/$USER/milvus/db:/var/lib/milvus/db \
-v /home/$USER/milvus/conf:/var/lib/milvus/conf \
-v /home/$USER/milvus/logs:/var/lib/milvus/logs \
-v /home/$USER/milvus/wal:/var/lib/milvus/wal \
milvusdb/milvus:0.10.2-gpu-d081520-8a2393

参考:

https://zhuanlan.zhihu.com/p/90812583

推荐 0
本文由 safa 创作,采用 知识共享署名-相同方式共享 3.0 中国大陆许可协议 进行许可。
转载、引用前需联系作者,并署名作者且注明文章出处。
本站文章版权归原作者及原出处所有 。内容为作者个人观点, 并不代表本站赞同其观点和对其真实性负责。本站是一个个人学习交流的平台,并不用于任何商业目的,如果有任何问题,请及时联系我们,我们将根据著作权人的要求,立即更正或者删除有关内容。本站拥有对此声明的最终解释权。

0 个评论

要回复文章请先登录注册