windows下yolov8训练环境配置

安装

anaconda3

conda create -n yolov8 python=3.8
## 进入yolov8 虚拟环境
activate yolov8 

pip install ultralytics

conda install chardet

安装pytorch、cuda

https://pytorch.org/get-started/locally/

image-20240524174628802

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

安装onnx格式支持

conda install onnx

导出pt模型为onnx格式:

yolo export model=best.pt format=onnx

https://docs.ultralytics.com/tasks/detect/#models

训练:

yolo detect train data=my_coco.yaml model=yolov8s.pt epochs=150 imgsz=640 batch=64 workers=0 device=0

本文作者:一颗西瓜糖

本文链接:https://www.cnblogs.com/ykxgt/p/18361059

版权声明:本作品采用知识共享署名-非商业性使用-禁止演绎 2.5 中国大陆许可协议进行许可。

posted @   一颗xi瓜糖  阅读(145)  评论(0编辑  收藏  举报
点击右上角即可分享
微信分享提示
评论
收藏
关注
推荐
深色
回顶
收起