Hi @art I can confirm that for an ARM64 platform with NVIDIA device we have access to (AWS G5g) we can fully compile and run the test-suite without any memory issues:
May I suggest clearing out your existing python environment and starting again from fresh with
python -m venv pyenv
source ./pyenv/bin/activate
python -m pip install pennylane cuquantum ninja cmake
git clone https://github.com/PennyLaneAI/pennylane-lightning-gpu
cd pennylane-lightning-gpu
python -m pip install -e .
I suspect if your file does not run successfully after this, there may be an issue with cuQuantum on the Xavier platform, and will not be something we can fully support.
Please feel free to let us know if this is the case, and we can try revisit this in future after a discussion with the cuQuantum team.