Cannot install pytorch3d on Jetson nano

[sonnabanana877]

Hi,
I would like to install pytorch3d on Jetson Nano but it failed.

Problem

The install command I tried
pip install 'git+https://github.com/facebookresearch/pytorch3d.git'

and got this error:

cub/cub.cuh: No such file or directory

Environment

nvcc -V

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2021 NVIDIA Corporation
Built on Sun_Feb_28_22:34:44_PST_2021
Cuda compilation tools, release 10.2, V10.2.300
Build cuda_10.2_r440.TC440_70.29663091_0

cat /etc/nv_tegra_release

# R32 (release), REVISION: 7.3, GCID: 31982016, BOARD: t210ref, EABI: aarch64, DATE: Tue Nov 22 17:30:08 UTC 2022

I found cub/cub.cuh at here and download it but I’m not sure how can I include it when the pip install process.

GitHub - NVlabs/cub: THIS REPOSITORY HAS MOVED TO github.com/nvidia/cub,...

I would be happy to help me.

[AastaLLL]

Hi,

Please check the below directory.
It should comes with CUDA library already.

/usr/local/cuda/targets/aarch64-linux/include/

Thanks.