Trouble about running the NMS function

FeiYull / TensorRT-Alpha

🔥🔥🔥TensorRT for YOLOv8、YOLOv8-Pose、YOLOv8-Seg、YOLOv8-Cls、YOLOv7、YOLOv6、YOLOv5、YOLONAS......🚀🚀🚀CUDA IS ALL YOU NEED.🍎🍎🍎

GNU General Public License v2.0

1.28k stars 198 forks source link

Trouble about running the NMS function #8

Closed anqi666 closed 1 year ago

anqi666 commented 1 year ago

I am having trouble running the NMS function in the TensorRT-Alpha repository. I am getting an error message related to CUDA support, and I am not sure what the issue is. Can you please provide guidance on how to resolve this issue? Thank you.

FeiYull commented 1 year ago

@anqi666 hi, TensorRT8.4.2.4 support CUDA 11.0, 11.1, 11.2, 11.3, 11.4, 11.5, 11.6 and 11.7. You can use the following command to verify whether the driver and cuda are installed normally： nvidia-smi nvcc -V reference doc : https://github.com/FeiYull/TensorRT-Alpha/blob/main/Install_For_Ubuntu18.04/Install_For_Ubuntu18.04.md