Problem in installing torch in Jetson AGX Orin

I’m trying to install PyTorch in my Jetson AGX Orin (Package: nvidia-jetpack Version: 5.0.1-b118) by following the steps provided in PyTorch for Jetson

  1. I choose to install PyTorch v1.11.0 and executing the command pip3 install numpy torch-1.11.0-cp38-cp38-linux_aarch64.whl resulting in installing torch version 1.8.0 as shown in the image above.

Python version: 3.8.10
pip3 version: 22.3.1

  1. when trying to import torch i get the following error

Hi,

Could you share the full command you used for installing PyTorch?
Please noted that the download link will be different according to the version you choose.

For example, the download command for v1.11.0 should look like as below:

$ wget https://nvidia.box.com/shared/static/ssf2v7pf5i245fk4i0q926hy4imzs2ph.whl -O  torch-1.11.0-cp38-cp38-linux_aarch64.whl

Thanks.

Hello AastaLLL, yes I’m sure that I changes that also, and I downloaded the right wheel (1.11), I was very attentive to every step to download the right version.

Hi,

We have tested the above command and it can work correctly.

$ python3
Python 3.8.10 (default, Nov 14 2022, 12:59:47)
[GCC 9.4.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.__version__
'1.11.0'
>>>

Here is the checksum info for your reference:

$ md5sum torch-1.11.0-cp38-cp38-linux_aarch64.whl
b9c21e288937d87175ba47b914f40809  torch-1.11.0-cp38-cp38-linux_aarch64.whl

Thanks.

1 Like

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