Jetson Agx Orin Problem while installing Pytorch

[avinashkrrai2]

I am using jetpack 5.1.2 with python 3.8 and I am trying to install pytorch 2.1.0 but pytorch 1.8.0 is getting installed and it is creating problem in the further process how to solve it.

[AastaLLL]

Hi,

Could you try to import the PyTorch and share the output of the below command with us?

$ python3
# import torch
# torch.__version__

Thanks.

[avinashkrrai2]

This is the output I am getting

[AastaLLL]

Hi,

It looks like the PyTorch in your environment has a dependency on cuBLAS 10.
So the installed version is 1.8.

Could you remove the package and re-install the PyTorch 2.1.nv23.06 package again?
Below is the checksum value of PyTorch v2.1 for your reference:

$ md5sum torch-2.1.0a0+41361538.nv23.06-cp38-cp38-linux_aarch64.whl
0063b1373e339e47995afd093a3cde87  torch-2.1.0a0+41361538.nv23.06-cp38-cp38-linux_aarch64.whl

Thanks.

[avinashkrrai2]

Thanks for your support I have finally installed torch2.1.0 and torchvision0.16.1 and now i want to install ultralytics yolov8 without disturbing the version of torch and torchvision how can I do it

[AastaLLL]

Hi

Ultraylytics can be installed via pip directly.
Do you meet any issues when using it?

Thanks.