laugh12321 / TensorRT-YOLO

🚀 你的YOLO部署神器。TensorRT Plugin、CUDA Kernel、CUDA Graphs三管齐下,享受闪电般的推理速度。| Your YOLO Deployment Powerhouse. With the synergy of TensorRT Plugins, CUDA Kernels, and CUDA Graphs, experience lightning-fast inference speeds.
https://github.com/laugh12321/TensorRT-YOLO
GNU General Public License v3.0
720 stars 81 forks source link

[Help]: Failed to detect a default CUDA architecture. #47

Closed cll-git closed 1 month ago

cll-git commented 1 month ago

Thanks for you contributions for the community.

However, I meat an error " Failed to detect a default CUDA architecture " when I run command " cmake -DTENSORRT_PATH=/usr/src/tensorrt ..".

Do you have any solution on it?

laugh12321 commented 1 month ago

@cll-git 先确定你有没有安装 CUDA Toolkit 并配置了环境变量, 如果按照上述步骤操作还是有问题可以参考以下链接解决。

https://www.cnblogs.com/metaz/p/16919028.html https://blog.csdn.net/qq_39779233/article/details/140281518

cll-git commented 1 month ago

我用的是jetson上的cuda,同时按照NVIDIA的官方教程,确认正确安装了cuda了的

------------------ 原始邮件 ------------------ 发件人: @.>; 发送时间: 2024年10月17日(星期四) 下午5:52 收件人: @.>; 抄送: @.>; @.>; 主题: Re: [laugh12321/TensorRT-YOLO] [Help]: Failed to detect a default CUDA architecture. (Issue #47)

@cll-git 先确定你有没有安装 CUDA Toolkit, 如果安装了还是有这个报错可以参考这个链接解决。https://www.cnblogs.com/metaz/p/16919028.html

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you were mentioned.Message ID: @.***>

laugh12321 commented 1 month ago

@cll-git 看看输入 nvcc --version 是否正常,加上 -DCMAKE_CUDA_COMPILER=/usr/local/cuda/bin/nvcc 试试。

cll-git commented 1 month ago

感谢,已解决。是由于jetson设备的CUDA是jetpack自动安装的,所以我忘记在bashrc里面添加cuda的路径了,添加上去就正常了

------------------ 原始邮件 ------------------ 发件人: @.>; 发送时间: 2024年10月17日(星期四) 下午5:56 收件人: @.>; 抄送: @.>; @.>; 主题: Re: [laugh12321/TensorRT-YOLO] [Help]: Failed to detect a default CUDA architecture. (Issue #47)

@cll-git 看看输入 nvcc --version 是否正常,加上 -DCMAKE_CUDA_COMPILER=/usr/local/cuda/bin/nvcc 试试。

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you were mentioned.Message ID: @.***>

laugh12321 commented 1 month ago

@cll-git 没事,问题解决了就行。这个issue就关闭了。