Jetpack 7.2. Jetson Orin Nano Super (8gb) Pytorch for Compute Capability 8.7

Robotics & Edge Computing Jetson Systems Jetson Orin Nano
, ,
4

Hi,

This message is a harmless warning:

/home/jetson1/.local/lib/python3.12/site-packages/torch/cuda/init.py:422: UserWarning: Found GPU0 Orin which is of compute capability (CC) 8.7.
The following list shows the CCs this version of PyTorch was built for and the hardware CCs it supports:

You can install upstream PyTorch on JetPack 7.2+Orin with the below command:

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

Below is the command for ignoring the warning for your reference:

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

Please find the comment below for details:

Thanks.