版本:https://milvus.io/docs/v0.10.2/milvus_docker-gpu.md
确认后台是否运行了Docker daemon
sudo docker info # 确认后台是否运行了Docker daemon
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服务
创建并启动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
重启电脑后:
解决 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
发现19530端口连不上,说明docker停了,可用下列命令查看
sudo docker info # 确认后台是否运行了Docker daemon
sudo service docker status # 查看docker服务状态
sudo systemctl start docker # 启动docker
sudo docker ps -a # 查看所有的容器
sudo docker logs milvus_gpu_0.10.2 # 查看docker服务日志
sudo docker top milvus_gpu_0.10.2 #查看进程信息
sudo docker stats milvus_gpu_0.10.2 #查看容器状态
sudo docker system df # 查看镜像、容器、数据卷所占用的空间
sudo docker image ls -a # 列出已下载镜像参考https://yeasy.gitbook.io/docker_practice/image/list
sudo docker image ls -f dangling=true # 显示悬虚镜像
sudo docker image prune # 删除悬虚镜像
重新启动有两种方法
方法一:删除原容器,用run命令重新创建 启动
sudo service docker restart # 启动docker
sudo docker rm milvus_gpu_0.10.2 # 删除已有的milvus容器
# 创建并启动容器
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
方法二:用start再次启动已有的容器
sudo service docker restart # 启动docker
sudo docker start milvus_gpu_0.10.2
参考: