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!