首页 最新 热门 推荐

  • 首页
  • 最新
  • 热门
  • 推荐
2025年6月8日 星期日 2:08am

win10下cuda12.1 +troch2.4.1+vs2022环境下编译安装flash-attn

  • 25-04-25 06:21
  • 3054
  • 9987
blog.csdn.net

步骤一 下载项目
先下载 https://github.com/Dao-AILab/flash-attention,然后在conda环境中进入项目目录

步骤二 安装依赖项
执行以下命令,安装cutlass库,该库为编译flash-attn的必须依赖

conda update --force conda 
conda install conda-forge::cutlass
  • 1
  • 2

conda下查找: https://anaconda.org/conda-forge/

步骤三 设置include目录
在flash-attention项目下,setup.py,第300行,添加 "c:/Users/Administrator/miniconda3/Library/include",这里为自己conda环境下的include目录,该命令下有conda安装的库文件,如cutlass。
在这里插入图片描述
具体修改效果如下
在这里插入图片描述

步骤四 突破vs2022与cuda版本限制
在cuda v12.1\include\crt\host_config.h下,表明不支持vs2022,强制进行编译会输出以下报错

error -- unsupported Microsoft Visual Studio version! Only the versions between 2017 and 2022 (inclusive) are supported! The nvcc flag '-allow-unsupported-compiler' can be used to override this version check; however, using an unsupported host compiler may cause compilation failure or incorrect run time execution. Use at your own risk
  • 1

参考:https://blog.csdn.net/lishiyu93/article/details/114599859,修改_MSC_VER的版本范围 原先是1940,被博主修改成了2940
在这里插入图片描述

在C:\Program Files\Microsoft Visual Studio\2022\Community\VC\Tools\MSVC\14.41.34120\include\yvals_core.h中,表明只支持cuda12.4以上的版本,将原来的代码参考 http://iyenn.com/rec/1823770.html ,修改为 #if __CUDACC_VER_MAJOR__ < 10 || (__CUDACC_VER_MAJOR__ == 10 && __CUDACC_VER_MINOR__ < 1)

在这里插入图片描述

步骤五 进行编译
执行 python setup.py build,进行编译
在这里插入图片描述

步骤六 生成whl文件
执行python .\setup.py bdist_wheel生成whl文件,命令行输出如下
在这里插入图片描述
生成的whl文件在dist目录下
在这里插入图片描述
博主编译好的whl文件下载地址: https://download.csdn.net/download/a486259/89839302

步骤七 安装包
执行 pip install .\flash_attn-2.6.3-cp38-cp38-win_amd64.whl 安装包
在这里插入图片描述

注:本文转载自blog.csdn.net的万里鹏程转瞬至的文章"https://blog.csdn.net/a486259/article/details/142695690"。版权归原作者所有,此博客不拥有其著作权,亦不承担相应法律责任。如有侵权,请联系我们删除。
复制链接
复制链接
相关推荐
发表评论
登录后才能发表评论和回复 注册

/ 登录

评论记录:

未查询到任何数据!
回复评论:

分类栏目

后端 (14832) 前端 (14280) 移动开发 (3760) 编程语言 (3851) Java (3904) Python (3298) 人工智能 (10119) AIGC (2810) 大数据 (3499) 数据库 (3945) 数据结构与算法 (3757) 音视频 (2669) 云原生 (3145) 云平台 (2965) 前沿技术 (2993) 开源 (2160) 小程序 (2860) 运维 (2533) 服务器 (2698) 操作系统 (2325) 硬件开发 (2492) 嵌入式 (2955) 微软技术 (2769) 软件工程 (2056) 测试 (2865) 网络空间安全 (2948) 网络与通信 (2797) 用户体验设计 (2592) 学习和成长 (2593) 搜索 (2744) 开发工具 (7108) 游戏 (2829) HarmonyOS (2935) 区块链 (2782) 数学 (3112) 3C硬件 (2759) 资讯 (2909) Android (4709) iOS (1850) 代码人生 (3043) 阅读 (2841)

热门文章

101
推荐
关于我们 隐私政策 免责声明 联系我们
Copyright © 2020-2025 蚁人论坛 (iYenn.com) All Rights Reserved.
Scroll to Top