Installing on Jetson Nano/Xavier NX (ARM64)

[ApluUalberta]

Hi there,
I’m currently trying to set up Pytorch Lightning on Jetson Nano/Jetson Xavier NX by building from source. So far, I have tried following this thread here:
https://github.com/PyTorchLightning/pytorch-lightning/issues/695

The requirements.txt has been changed and no longer has torchvision and scikit-learn as one of the requirements. However, it seems to seek a torch version>=1.4 as a result of torchmetrics>=0.2.0 (within requirements.txt). My Jetson even has torch 1.8.0 and torchvision in its pip3 package manager. I was wondering if anyone has successfully set Pytorch Lightning up on ARM64's new requirement layout. Thanks!

My pip install attempt:

my pip install requirements attempt:

my pip list:

Is there something I’m missing? I have also tried running setup.py to no avail. Thanks!

What's your environment?

• OS: Linux
• Hardware: Jetson Nano
• Packaging: Pip
• Processor: ARM