首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >专栏 >[深度学习][原创]yolact编译DCNv2错误解决方法

[深度学习][原创]yolact编译DCNv2错误解决方法

作者头像
云未归来
发布2025-07-18 17:53:46
发布2025-07-18 17:53:46
610
举报

项目地址:https://github.com/dbolya/yolact

使用yolact编译DCNv2提示

/media/howso/f/lu/myproj/yolact-master/external/DCNv2/src/cuda/dcn_v2_cuda.cu(107): error: identifier "THCState_getCurrentStream" is undefined

/media/howso/f/lu/myproj/yolact-master/external/DCNv2/src/cuda/dcn_v2_cuda.cu(279): error: identifier "THCState_getCurrentStream" is undefined

/media/howso/f/lu/myproj/yolact-master/external/DCNv2/src/cuda/dcn_v2_cuda.cu(324): error: identifier "THCudaBlas_Sgemv" is undefined

3 errors detected in the compilation of "/media/howso/f/lu/myproj/yolact-master/external/DCNv2/src/cuda/dcn_v2_cuda.cu". error: command '/usr/local/cuda-11.0/bin/nvcc' failed with exit status 1 我的环境是:

ubuntu18.04

RTX3090

torch==1.7.0

torchvision==0.8.1

cuda11.0+cudnn8.1.1

原因是DCNv2对于RTX30系列支持不好,需要重新去https://github.com/MatthewHowe/DCNv2

下载,而且这样还不够需要在bash make.sh之前

export TORCH_CUDA_ARCH_LIST=7.5

虽然RTX3090算力是8.6但是还是会提示错误,设置7.5就可以

本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2022-01-07,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档