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