Pytorch canot detect GPU jetson orin nano jetpack 6.1

Robotics & Edge Computing Jetson & Embedded Systems Jetson Orin Nano
1

HI, my GPU in jetson orin nano with jetpack 6.1 canot be detected by pytorch

after installing jetpack 6.1 i try to check cuda with pytorch in python but it return false,
and then i try to install pytorch from pytorch start localy, but there is not have torch for cuda 12.6 which is used by my jetson.

4

Hello,

You can visit this site to download the Torch and TorchVision wheels compatible with CUDA 12.6.

Thanks

1 Like
6

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