windows 训练yolov26官方数据集

qq762118222026-04-30 10:42

一、安装Anaconda3-2025.12-2-Windows-x86_64.exe

下载地址：https://repo.anaconda.com/archive/

1.1. 升级本机显卡驱动

参考：https://blog.csdn.net/sz76211822/article/details/160592931

二、创建环境

2.1 查看当前环境

打开Anaconda Prompt 输入

复制代码

conda info -e

打印：

2.2 安装python3.10

按官网所说安装

2.3 创建环境

复制代码

conda create -n yolo python=3.10

2.4 激活环境

三、安装pytorch

复制代码

conda install pytorch=2.5.1 torchvision=0.20.1 torchaudio=2.5.1 pytorch-cuda=12.4  -c pytorch -c nvidia -y

conda install ultralytics matplotlib 

conda install opencv=4.13.0

四、查看是否支持GPU

五、下载ultralytics 代码

复制代码

git clone https://github.com/ultralytics/ultralytics ./ultralytics

下载完成后，在当前目录打开

设置解释器

新增解释器

选择 yolo 的路径，然后点击"OK"

六、训练

在当前目录添加一个train.py

复制代码

from ultralytics import YOLO

def main():
    model = YOLO("yolo26n.pt")  # 或 yolov8n.pt
    train_results = model.train(
        data="coco8.yaml",
        epochs=100,
        imgsz=640,
        device="0",
        workers=2,  # 可保留，但需主函数保护
        batch=8,
    )

if __name__ == '__main__':
    main()

然后点击运行即可