Exporting a yolov5s best.pb Model to .engine Format (TensorRT)

Thanks @andres.artavia,

I finally managed to install PyTorch.

After running the command python setup.py install, I had to go to the /dist/ folder and execute pip install torch-1.8.0a0+37c1f4a-cp37-cp37m-linux_aarch64.whl.

After that, I also succeeded in installing TorchVision 0.9.0, but now I’m facing the issue of not having a valid candidate for tensorrt.whl. Everything I’ve read suggests that I need to compile TensorRT from source, but I don’t know how to do it*, and I’m not even sure which version would be compatible with my architecture (aarch64) and CUDA 10.2.

*I saw here that I can build it from the source but the info inside the github readme is not so clear for me. I tried it but i failed.

If someone can give me more detailed steps I would be really glad about it.
For example, i don’t know if TensorRT 8.2 is suitable for python 3.7, pytorch 1.8 or I will need to move to another TensorRT version.

Thank you very much!