在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


  • 0
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值