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

when GPU Framwork version number is less than 61, compile error #8

Closed
guods opened this issue Mar 5, 2020 · 6 comments
Closed

when GPU Framwork version number is less than 61, compile error #8

guods opened this issue Mar 5, 2020 · 6 comments

Comments

@guods
Copy link

guods commented Mar 5, 2020

  1. complie error
    more than one conversion function from "const __half" to a built-in type applies; on line 44 PReLUPlugin.cu
  • OS: windows10, cuda10.0, cudnn7.5.0
  • TensorRT version: TensorRT5.1.5.0
@guods
Copy link
Author

guods commented Mar 5, 2020

get the same error with TnesorRT 7.0.1.1

@zerollzeng
Copy link
Owner

that's because when graphic card's sm version less than 6.0, they don't support fp16, and my custom plugin has a fp16 implementation, which graphic card you are using now?

@zerollzeng
Copy link
Owner

@guods
Copy link
Author

guods commented Mar 5, 2020

GTX TiTan X(52), but I run the demo https://github.com/lewes6369/TensorRT-Yolov3 on windows10 with TensorRT5.1.5.0 , cuda10.0 and cudnn7.5.0, the result is normal.

@guods
Copy link
Author

guods commented Mar 5, 2020

Thanks for your reply, the problem has been solved on TiTan X.

@guods guods closed this as completed Mar 5, 2020
@zerollzeng
Copy link
Owner

that's becuase I disable sm version less than 6.0. if you want to make it run, you can go to
https://github.com/zerollzeng/tiny-tensorrt/blob/69e64d10270b518c1356b50d9dff1553e3d8c1ce/cmake/CUDA_utils.cmake#L17
then comment some lines related to fp16 like
https://github.com/zerollzeng/tiny-tensorrt/blob/69e64d10270b518c1356b50d9dff1553e3d8c1ce/plugin/PReLUPlugin/PReLUPlugin.cu#L184
I think that would solve your problem

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