Orin AGX, JP 7.2, Pytorch and sm_87 support?

Robotics & Edge Computing Jetson Systems Jetson AGX Orin
,
6

Hi,

Do you install the upstream PyTorch?

$ pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu132

If so, it is a harmless warning. You can turn it off with the following function:

import warnings
warnings.filterwarnings("ignore", message=".*Found GPU.*compute capability.*")

The package has been built with sm_80 and can be run on the Orin without triggering JIT compiling:

An example of the latter is that code compiled for the target sm_80 will run on all other CC 8.x GPUs, such as sm_86 or sm_89.*

This means the upstream packages contain the kernel code that can directly run on Orin.
The warning can be turned off without any performance or capability issues.

Thanks.