yolov12 windows环境搭建及训练

叽里呱啦Jade

已于 2025-02-25 11:21:13 修改

阅读量929

点赞数 6

文章标签： windows YOLO 深度学习

于 2025-02-25 11:18:50 首次发布

本文链接：https://blog.csdn.net/weixin_43269340/article/details/145847167

版权

yolov12环境搭建及训练

yolo12最新发布出来，Windows搭建环境遇到一点问题，因为官方发布的是linux版本，导致Windows按照步骤的包不一样，特此记录一下。

一、创建yolov12环境

打开anaconda的prompt,创建一个yolov12的环境；

conda create -n yolov12 python==3.11

等待安装成功，激活环境

conda activate yolov12

二、下载yolov12安装包

官方链接: link

我这里是直接Download ZIP，解压并进入文件夹

cd yolov12-main

在这里插入图片描述
可以看到里面有一个requirments.txt，打开文件可以看到，其中有一个轮子是：
flash_attn-2.7.3+cu11torch2.2cxx11abiFALSE-cp311-cp311-linux_x86_64.whl
linux版本，所以Windows安装不能直接安装requirments.txt，需要改成windows版本；

因为官方版本只有linux版本，在网上找到大佬发布的编译好的windows轮子，参考下面版本
链接：link
在这里插入图片描述
我电脑上的CUDA为12.6版本，python3.11，选择对应的pytorch2.5.1版本，下载
flash_attn-2.7.4.post1+cu124torch2.5.1cxx11abiFALSE-cp311-cp311-win_amd64.whl版本，准备好

三、安装需要的包

pytorch官网：[https://pytorch.org/get-started/previous-versions/]
首先安装pytorch，可以参考之前的安装流程，在官网选择版本，安装好pytorch；

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

安装过程可能有点慢，需要耐心等待。

安装其他环境

pip install -e .

安装其他环境

安装好pytorch之后，继续安装flash_attn

pip install flash_attn-2.7.4.post1+cu124torch2.5.1cxx11abiFALSE-cp311-cp311-win_amd64.whl

耐心等待安装好之后，查看requirments.txt的内容，安装所需要的包，我怕出现不兼容的现象，所有后面模糊掉版本，逐个安装也可以，至此环境就安装成功了。