[python]windows上安装yolov12环境

爱谁谁
发布: 2025-08-28 10:14:20
原创
377人浏览过

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页面查找是否有适用于Windows的
win_amd64.whl
登录后复制
文件。可惜的是,官方并未提供Windows版本的预编译包,仅支持Linux。因此,若想在Windows上运行,要么寻找第三方提供的whl文件,要么自行从源码编译。

尝试从源码编译有一定难度,对环境要求较高,因此更推荐寻找已编译好的Windows版本whl包。例如可以从gitee.com/FIRC/flash_attn_chinese_mirror获取,当然你也可以搜索其他可信来源。

我本地使用的是Python 3.10环境,当然你也可以选择Python 3.11或其他版本。

创建并激活conda环境:

立即学习Python免费学习笔记(深入)”;

conda create -n yolov12 python=3.10
conda activate yolov12
登录后复制

我下载的flash-attn版本为:

flash_attn-2.7.4+cu124torch2.5.1cxx11abiFALSE-cp310-cp310-win_amd64.whl
登录后复制

为确保版本兼容,我同步调整了PyTorch版本,安装命令如下:

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 .
登录后复制

安装完成后,进行图片推理测试:

Cutout老照片上色
Cutout老照片上色

Cutout.Pro推出的黑白图片上色

Cutout老照片上色 20
查看详情 Cutout老照片上色
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:naconda3envsyolo12libsite-packages	immmodelslayers__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:nimal.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 runsdetectpredict
登录后复制

看到提示“FlashAttention is not available on this device”,怀疑flash-attn未正确安装。于是进入Python环境手动导入测试:

[python]windows上安装yolov12环境

手动import测试:

[python]windows上安装yolov12环境

导入成功,说明flash-attn安装无误。问题出在

torch.cuda.get_device_capability()[0] >= 8
登录后复制
这一条件不满足。我的显卡是RTX 2070,计算能力为7.5,低于8.0,而flash-attn可能仅对算力8.0及以上设备启用优化功能。因此该警告可忽略。

最终效果如下:

[python]windows上安装yolov12环境

环境配置成功,yolov12可在Windows上正常运行!

以上就是[python]windows上安装yolov12环境的详细内容,更多请关注php中文网其它相关文章!

Windows激活工具
Windows激活工具

Windows激活工具是正版认证的激活工具,永久激活,一键解决windows许可证即将过期。可激活win7系统、win8.1系统、win10系统、win11系统。下载后先看完视频激活教程,再进行操作,100%激活成功。

下载
来源:php中文网
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
最新问题
开源免费商场系统广告
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板
关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新 English
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送
PHP中文网APP
随时随地碎片化学习

Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号