[python]windows上安装yolov12环境

yolov12出来了，地址github.com/sunsmarterjie/yolov12，咱们看看怎么在windows上把环境安装一下首先看看官方安装流程：

wget https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.3/flash_attn-2.7.3+cu11torch2.2cxx11abiFALSE-cp311-cp311-linux_x86_64.whl
conda create -n yolov12 python=3.11
conda activate yolov12
pip install -r requirements.txt
pip install -e .

很明显官方只能在linux上安装，因为flash_attn这个是linux上的whl，然后去flash-attention源码页面的release找下有没有win_amd64.whl。很遗憾这个官方都是直接提供linux上的whl，windows上官方不直接支持。因此要么从其他途径找windows上的whl要么从源码编译。我试了下从源码开始有点难度，所以还是从其他地方找到whl，比如gitee.com/FIRC/flash_attn_chinese_mirror，当然您也可以去其他地址找一下。

我安装的是python3.10版本环境，大家也可以安装python3.11或者其他环境

conda create -n yolov12 python=3.10
conda activate yolov12

我下载了flash_attn-2.7.4+cu124torch2.5.1cxx11abiFALSE-cp310-cp310-win_amd64.whl

为了匹配对应版本我更换了torch版本。安装如下

pip install torch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 --index-url https://download.pytorch.org/whl/cu124

注释掉官方requirements.txt里面torch和flash-attn

#torch==2.5.1 
#torchvision==0.20.1
#flash_attn-2.7.3+cu11torch2.2cxx11abiFALSE-cp311-cp311-linux_x86_64.whl
timm==1.0.14
albumentations==2.0.4
onnx==1.14.0
onnxruntime==1.15.1
pycocotools==2.0.7
PyYAML==6.0.1
scipy==1.13.0
onnxslim==0.1.31
onnxruntime-gpu==1.18.0
gradio==4.44.1
opencv-python==4.9.0.80
psutil==5.9.8
py-cpuinfo==9.0.0
huggingface-hub==0.23.2
safetensors==0.4.3
numpy==1.26.4

pip install -r requirements.txt
pip install -e . 

安装完成，接下来测试图片

yolo task=detect mode=predict model=weights/yolov12n.pt source=E:\animal.jpg save=true

FlashAttention is not available on this device. Using scaled_dot_product_attention instead.
D:\anaconda3\envs\yolo12\lib\site-packages\timm\models\layers\__init__.py:48: FutureWarning: Importing from timm.models.layers is deprecated, please import via timm.layers
  warnings.warn(f"Importing from {__name__} is deprecated, please import via timm.layers", FutureWarning)
Ultralytics 8.3.63 🚀 Python-3.10.16 torch-2.5.1+cu124 CUDA:0 (NVIDIA GeForce RTX 2070 Super, 8192MiB)
YOLOv12n summary (fused): 352 layers, 2,590,824 parameters, 0 gradients, 6.5 GFLOPs

image 1/1 E:\animal.jpg: 640x640 1 zebra, 1 giraffe, 16.0ms
Speed: 4.0ms preprocess, 16.0ms inference, 91.0ms postprocess per image at shape (1, 3, 640, 640)
Results saved to runs\detect\predict

上面提示FlashAttention is not available on this device,难道flash-attn没安装好？于是我定位源码

手动import

ok说明flash-attn安装没有问题,只能torch.cuda.get_device_capability()[0] >= 8不符合要求了。我电脑是RTX2070显卡，算力7.5，没有超过8，看起来flash-attn专门为算力>=8设计的模块？不过警告不用管直接查看

安装完成！