Torch not compiled with cuda enabled over Jetson Xavier Nx

[vashisht.akshat.rn.04]

I am having Jetson Xavier Nx with jetpack 4.6.

I am having PyTorch 1.8.0 and cuda 10.2
When I try to import pytorch, it successfully imports the pytorch module, But when I run the code for achieving stereo vision depth estimation it gives an error that → Torch not compiled with cuda enabled.

After that I have visited to → Previous PyTorch Versions | PyTorch

# CUDA 10.2
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=10.2 -c pytorch

and applied the commands of it but the error remains the same.
Please tell me how to resolve this error as soon as possible.

[AastaLLL]

Hi,

For Jetson, please install the PyTorch package from the below topic:

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…

If you want a python v3.7 package, please build it from the source with the instructions in the same topic.

Thanks.