How do I install pytorch 1.5.0 on the jetson nano devkit?

Hello,

How do I install pytorch 1.5.0 on the jetson nano devkit?

I am asked because pytorch (1.7.0) does not work in the latest version, so I need to install 1.5.0, but I am told that pytorch 1.5.0 cannot be installed in the latest version of jetpack.
Thank you.

Hi,

PyTorch can work on JetPack4.4.1.
Please find the package in this document:

Thanks.

1 Like

Hello, @kayccc @AastaLLL

pip3 install numpy torch-1.5.0-cp36-cp36m-linux_aarch64.whl
Collecting numpy
Downloading https://files.pythonhosted.org/packages/c5/63/a48648ebc57711348420670bb074998f79828291f68aebfff1642be212ec/numpy-1.19.4.zip (7.3MB)
100% |████████████████████████████████| 7.3MB 71kB/s
Processing ./torch-1.5.0-cp36-cp36m-linux_aarch64.whl
Collecting future (from torch==1.5.0)
Downloading https://files.pythonhosted.org/packages/45/0b/38b06fd9b92dc2b68d58b75f900e97884c45bedd2ff83203d933cf5851c9/future-0.18.2.tar.gz (829kB)
100% |████████████████████████████████| 829kB 558kB/s
Collecting dataclasses (from torch==1.5.0)
Downloading https://files.pythonhosted.org/packages/e1/d2/6f02df2616fd4016075f60157c7a0452b38d8f7938ae94343911e0fb0b09/dataclasses-0.7-py3-none-any.whl
Collecting typing-extensions (from torch==1.5.0)
Downloading https://files.pythonhosted.org/packages/60/7a/e881b5abb54db0e6e671ab088d079c57ce54e8a01a3ca443f561ccadb37e/typing_extensions-3.7.4.3-py3-none-any.whl
Building wheels for collected packages: numpy, future
Running setup.py bdist_wheel for numpy … done
Stored in directory: /home/realwave/.cache/pip/wheels/b1/d6/2f/b576b6c983652fa9c02be4c2a4c3896e9540c3b5dc85439986
Running setup.py bdist_wheel for future … done
Stored in directory: /home/realwave/.cache/pip/wheels/8b/99/a0/81daf51dcd359a9377b110a8a886b3895921802d2fc1b2397e
Successfully built numpy future
Installing collected packages: numpy, future, dataclasses, typing-extensions, torch
Successfully installed dataclasses-0.7 future-0.18.2 numpy-1.19.4 torch-1.7.0 typing-extensions-3.7.4.3

I need to install pytorch version 1.5.0
Install Jetpack 4.4 DP
I got torch-1.5.0~.whl and installed it
Version 1.7.0 has been installed.

Please check.

Thank you.

Hi,

pyTorch v1.5.0 is for JetPack4.4 DP rather than JetPack 4.4.1.
If you prefer v1.5.0, please build it from source with the instructions shared in the topic above.

Thanks.

1 Like

Hello,

After installing your jetpack4.4 DP
I received and installed pytorch v1.5.0 whl.
I am not using Jetpack4.4.1.

cat /etc/nv_tegra_release
R32 (release), REVISION: 4.2, GCID: 20074772, BOARD: t210ref, EABI: aarch64, DATE: Thu Apr 9 01:22:12 UTC 2020

pip3 install numpy torch-1.5.0-cp36-cp36m-linux_aarch64.whl

============== install process =====================

pip3 install numpy torch-1.5.0-cp36-cp36m-linux_aarch64.whl
Collecting numpy
Downloading https://files.pythonhosted.org/packages/c5/63/a48648ebc57711348420670bb074998f79828291f68aebfff1642be212ec/numpy-1.19.4.zip (7.3MB)
100% |████████████████████████████████| 7.3MB 71kB/s
Processing ./torch-1.5.0-cp36-cp36m-linux_aarch64.whl
Collecting future (from torch==1.5.0)
Downloading https://files.pythonhosted.org/packages/45/0b/38b06fd9b92dc2b68d58b75f900e97884c45bedd2ff83203d933cf5851c9/future-0.18.2.tar.gz (829kB)
100% |████████████████████████████████| 829kB 558kB/s
Collecting dataclasses (from torch==1.5.0)
Downloading https://files.pythonhosted.org/packages/e1/d2/6f02df2616fd4016075f60157c7a0452b38d8f7938ae94343911e0fb0b09/dataclasses-0.7-py3-none-any.whl
Collecting typing-extensions (from torch==1.5.0)
Downloading https://files.pythonhosted.org/packages/60/7a/e881b5abb54db0e6e671ab088d079c57ce54e8a01a3ca443f561ccadb37e/typing_extensions-3.7.4.3-py3-none-any.whl
Building wheels for collected packages: numpy, future
Running setup.py bdist_wheel for numpy … done
Stored in directory: /home//.cache/pip/wheels/b1/d6/2f/b576b6c983652fa9c02be4c2a4c3896e9540c3b5dc85439986
Running setup.py bdist_wheel for future … done
Stored in directory: /home//.cache/pip/wheels/8b/99/a0/81daf51dcd359a9377b110a8a886b3895921802d2fc1b2397e
Successfully built numpy future
Installing collected packages: numpy, future, dataclasses, typing-extensions, torch
Successfully installed dataclasses-0.7 future-0.18.2 numpy-1.19.4 torch-1.7.0 typing-extensions-3.7.4.3

Thank you.

Hi,

Based on your log, the installation is succeed.
Do you meet any further issue when setting-up the pyTorch?

Thanks.

1 Like

Hello,

I explicitly installed version 1.5.0 with this command, but 1.7.0 was installed.

command : pip3 install numpy torch-1.5.0-cp36-cp36m-linux_aarch64.whl

Thank you.

Hi,

Could you get the checksum value of the downloaded package first?
In case you download the v1.7.0 package but mis-renaming it into v1.5.0.

Here is the result for comparison:

$ md5sum torch-1.5.0-cp36-cp36m-linux_aarch64.whl
1de1c0a70ad1556a5a37f9d7ac6de379  torch-1.5.0-cp36-cp36m-linux_aarch64.whl

Thanks.