windows 训练yolov26官方数据集
一、安装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.102.4 激活环境
三、安装pytorch
conda install pytorch=2.5.1 torchvision=0.20.1 torchaudio=2.5.1 pytorch-cuda=12.4 -c pytorch -c nvidia -yconda install ultralytics matplotlibconda 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()然后点击运行即可
