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. https://github.com/NVlabs/cub
