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

Issue with YOLOX-ROS TensorRT Conversion: CUDA Compatibility Error #44

Open
Buddies-as-you-know opened this issue Nov 20, 2023 · 3 comments

Comments

@Buddies-as-you-know
Copy link

Buddies-as-you-know commented Nov 20, 2023

Hello,

I am working on converting the YOLOX Tiny model to TensorRT format using the YOLOX-ROS package. However, I encountered a CUDA compatibility error during the conversion process. The command I used was:

./src/YOLOX-ROS/weights/tensorrt/convert.bash yolox_tiny 16

The script seemed to initiate correctly, but then it terminated with the following error message:

Cuda failure: forward compatibility was attempted on non supported HW
./src/YOLOX-ROS/weights/tensorrt/convert.bash: line 37:  8669 Aborted                 (core dumped) /usr/src/tensorrt/bin/trtexec --onnx=$SCRIPT_DIR/../onnx/$MODEL.onnx --saveEngine=$SCRIPT_DIR/$MODEL.trt --fp16 --verbose --workspace=$((1<<$TRT_WORKSPACE))

This error suggests a compatibility issue with my hardware and CUDA. I am wondering if you could provide some insights or guidance on how to resolve this issue. Here are some details about my setup:

Any help or suggestions to fix this compatibility issue would be greatly appreciated. Thank you for your time and assistance.

@Buddies-as-you-know Buddies-as-you-know changed the title Error Encountered in YOLOX-ROS TensorRT Conversion Issue with YOLOX-ROS TensorRT Conversion: CUDA Compatibility Error Nov 20, 2023
@Ar-Ray-code
Copy link
Owner

First of all you should tell me what environment you are using.

  • Ubuntu version
  • CUDA(-toolkit) version
  • TensorRT version
  • GPU (RTX4090? Jetson Orin?)

Errors during conversion are usually due to GPU, CUDA and TensorRT version mismatch.

@Buddies-as-you-know
Copy link
Author

Thanks for letting me know.

  • Ubuntu version:22.04
  • CUDA(-toolkit) version:11.4
  • TensorRT version:8.4
  • GPU :RTX3090

@Ar-Ray-code
Copy link
Owner

The following shows the environment in which I have worked at hand.
Please refer

  • Ubuntu22.04
  • GPU-driver : 525.89.02
  • CUDA : CUDA-11.8
  • TensorRT (trtexec) : v8.5.1
  • GPU : RTX3060

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