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:
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β¦
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.