在ubuntu上搭建基于docker的yolov5环境

文章目录

  • 相关环境准备
  • 启动容器、加载镜像、挂载数据集路径


相关环境准备

在官网下载docker

启动容器、加载镜像、挂载数据集路径

#拉取镜像
sudo docker pull ultralytics/yolov5:latest
docker run --name yolov5_test --ipc=host -it ultralytics/yolov5:latest
#可以指定端口映射进行启动

docker run  -v /home/lbh/ubuntu-home/datasets  --name yolov5_test -p 16022:22  -it yolov5_require:torch1.8-cu102-runtime-py3.9-ub18 
#指定端口映射、指定GPU模式进行启动
sudo docker run -v  --runtime=nvidia /home/lbh/yolov5_v6.1_test:/tmp/project  -p 16022:22  --name yolov5_gpu --gpus all -it ultralytics/yolov5:latest 
XXX
## 停止容器

docker stop yolov5_test

## 关闭容器

docker kill yolov5_test

## 重启容器

docker restart yolov5_test

## 删除容器

docker rm yolov5_test

# 运行检测代码

python detect.py --weights yolov5s.pt --source uniform/data/images

# 将检测结果从容器拷贝到宿主机

docker cp myyolov5:/usr/src/app/runs/detect D:\datasets\uniform\runs


你可能感兴趣的:(docker,ubuntu,容器)