Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

为啥不支持cuda11.0啊 #9

Open
chl916185 opened this issue Feb 11, 2022 · 1 comment
Open

为啥不支持cuda11.0啊 #9

chl916185 opened this issue Feb 11, 2022 · 1 comment

Comments

@chl916185
Copy link

我的显卡是rtx3090,只能使用cuda11.0以上的显卡,看您代码是在10.2上才可以运行啊,要修改哪里可以用cuda11.0啊

@lzh420202
Copy link
Owner

cuda11上面进行测试应该没问题,但是我们写的cuda扩展在cuda11版本上运行的结果和我们预想的结果(cuda10)不一致,所以我们并不推荐使用cuda11进行训练。这个问题需要完全基于cuda11的库重新构建fcosr的核心组件,会很耗时间。目前我们的工作重心并没有放在FCOSR上,而是试图设计一种更为高效的旋转目标检测器。你可以关注我们的另一个tensorrt推理库,我们后续所有的检测器都会想办法适配这个框架。
谢谢你的支持。

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants