windows编译DeformableDetr,cuda大于10.2
·
一、问题
在github上下载DeformableDetr代码,
https://github.com/fundamentalvision/Deformable-DETR
https://github.com/fundamentalvision/Deformable-DETR安装vs2019对应的一些c++文件,之前下载pycocotools已经装好了,
然后按照Compiling CUDA operators中进行编译操作,运行时发现没有找不到MultiScaleDeformableAttention库,直接pip安装找不到对应版本,貌似只有适合cuda10.2的,但我自身的cuda是11.3,又不想降低cuda版本。
二、操作
按照链接中所述的,在编译前先导入自己的CUDA路径:
pytorch 配置deformabledetr和referformer工程环境踩坑_部署deformable detr_咯吱咯吱咕嘟咕嘟的博客-CSDN博客pytorch 配置工程环境踩坑_部署deformable detr
https://blog.csdn.net/weixin_41469023/article/details/127605771在执行类似 sh ./make.sh操作前,输入
export CUDA_HOME=你的CUDA安装路径
我是用gitbash在models/ops里打开的,具体路径可以在环境变量里查看,

再次执行sh ./make.sh 就可以成功编译啦!!!
欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。
更多推荐



所有评论(0)