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

项目地址: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就可以

评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

FL1623863129

你的打赏是我写文章最大的动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值