Libtorch for AGX Xavier

Hello! I try to build Libtorch from source of PyTorch. My steps:

sudo nvpmodel -m 0
sudo jetson_clocks 

git clone --recursive http://github.com/pytorch/pytorch
cd pytorch

export MAX_JOBS=8 BUILD_TEST=0 USE_NCCL=0 TORCH_CUDA_ARCH_LIST="5.3;6.2;7.2" USE_SYSTEM_NCCL=0 USE_CUDA=1 USE_DISTRIBUTED=0 BUILD_TORCH=ON BUILD_BINARY=ON USE_NNPACK=0

pip install -U setuptools
pip install -r requirements.txt

pip install scikit-build
pip install ninja

python setup.py build

but it doesn’t build. What should I do to get build like that.

JetPack 4.2.
Thanks!

Hi,

We have a PyTorch prebuilt package. The package supports Nano, TX2 and Xavier with JetPack4.2.
https://devtalk.nvidia.com/default/topic/1049071/jetson-nano/pytorch-for-jetson-nano/

You can also build it from the source with the steps shared in the topic.

Thanks.