参考（官网教程）：
Train Custom Data - Ultralytics YOLOv8 Docs
0. 需要
0.1 自己的数据集（图片的形式）
0.2 科学上网
1.流程
1.1 git项目，安装依赖
git clone https://github.com/ultralytics/yolov5 # clone
cd yolov5
pip install -r requirements.txt # install
/**
如果要调用gpu，就把requirements.txt里的torch相关的包注释掉，然后去pytorch官网下载gpu版本的
例如 conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
如果已经有了cpu版本的又下载了gpu版本的，就会报错
OMP: Error #15: Initializing libiomp5md.dll, but found libiomp5md.dll already initialized. OMP: Hint This means that multiple copies of the OpenMP runtime have been linked into the program. That is dangerous, since it can degrade performance or cause
解决办法就是
pip uninstall torch
conda uninstall torch
conda uninstall libtorch
重新运行conda install pyto
1061
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
