Cannot install pytorch3d on Jetson nano

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


The install command I tried
pip install 'git+'

and got this error:

cub/cub.cuh: No such file or directory


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.

I would be happy to help me.


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



This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.