detectron2 c++ 部署笔记

本文记录了在Detectron2 C++部署中遇到的CUDA版本问题,包括compute 8.6与7.5不匹配的错误以及cublasStatus失败的解决尝试。在解决过程中,尝试了更新CUDA版本、修改编译参数和头文件路径,但部分问题通过升级到CUDA 11.1得到解决。同时提到了TensorRT 8.2版本的编译失败和C++中的E1455错误。
摘要由CSDN通过智能技术生成

目录

cuda版本:

cub特定版本

CUDA C/C++ 自定义命令,添加参数:  --extended-lambda 

链接器,输入库:

rotatedNms.cu 头文件

报错1:expecting compute 8.6 got compute 7.5

报错2:Assertion failed: cublasStatus == CUBLAS_STATUS_SUCCESS

tensorrt换8.2版本尝试失败


detectron2v04

cuda版本:

python只支持cuda10.2,

c++支持cuda10.2,11.1,后来发现11.1更合适。

cub特定版本

的目录拷贝到cuda根目录下:

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\include

如果不拷贝,则会报200多个错或者1千多个错。

CUDA C/C++ 自定义命令,添加参数:  --extended-lambda 

一个杠,两个杠有的可以

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

打赏作者

AI算法网奇

你的鼓励将是我创作的最大动力

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

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

打赏作者

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

抵扣说明:

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

余额充值