Do Jetson Xavier support PyTorch C++ API?

[abraham.pelz]

Do Jetson Xavier support PyTorch C++ API?

Thanks,
Avi

[AastaLLL]

Hi,

YES.

But please remember to add -Wl,–no-as-needed flag when building to load the library correctly.

Libtorch CUDA Problem

nvSample.zip (17.8 KB) Just run ./sampleLibtorch Source code is this- int main() { try { std::cout << torch::cuda::is_available() << std::endl; torch::Tensor tensor = at::tensor({ -1, 1 }, at::kCUDA); } catch (std::exception& ex) { std::cout << ex.what() << std::endl; } }

Thanks.

[abraham.pelz]

@AastaLLL

-Wl,–no-as-needed

Where do I need to add it in the make file?
And which C++ API version is supported for Jetson with JetPack 4.3?

[AastaLLL]

Hi,

You can check the Do Jetson Xavier support PyTorch C++ API? - #3 by AastaLLL for details.
There are some examples in the topic.

C++ pyTorch is supported for JetPack4.3
But please make sure you have installed the correct package which is built for JetPack4.3.

PyTorch for Jetson

Below are pre-built PyTorch pip wheel installers for Jetson Nano, TX1/TX2, Xavier, and Orin with JetPack 4.2 and newer. Download one of the PyTorch binaries from below for your version of JetPack, and see the installation instructions to run on your Jetson. These pip wheels are built for ARM aarch64 architecture, so run these commands on your Jetson (not on a host PC). You can also use the pre-built l4t-pytorch and l4t-ml container images and Dockerfiles. PyTorch pip wheels JetPack 5

PyTo…

Thanks.

[YoushaaMurhij]

Can you please provide some information about using Libtorch inside a Pytorch docker on Xavier?