I was able to run the Hello AI World lessons without issue on Jetpack 4.2. I have upgraded to Jetpack 4.4, and have a freshly flashed JP 4.4 card. Following Dusty’s tutorial, I do the following commands: $ sudo apt-get update $ sudo apt-get install git cmake libpython3-dev python3-numpy $ git clo…

OK, the install_pytorch.sh script has now been updated to install the PyTorch wheels for CUDA 10.2, sorry about that. Please refer to this post for more information: https://forums.developer.nvidia.com/t/pytorch-for-jetson-nano-version-1-5-0-now-available/72048/291

Hi, Do you apply extra pyTorch installation steps? Or just run the jetson_inference cmake and make command? Thanks.

I do the installation steps exactly as indicated in Hello AI worlds instructions, that is these steps: sudo apt-get update sudo apt-get install git cmake libpython3-dev python3-numpy git clone --recursive GitHub - dusty-nv/jetson-inference: Hello AI World guide to deploying deep-learning inference…

Confirming that this works now. Following the Hello AI World instructions with jetpack 4.4 results in working pytorch installation. Thanks!

I am following all installation steps as described in the Hello AI World lesson. In the lesson “Re-training on the Cat/Dog Dataset” when I try to “Re-training ResNet-18 Model” using the commands provided: $ cd jetson-inference/python/training/classification $ python train.py --model-dir=cat_dog ~…

[image] halimatif: python train.py --model-dir=cat_dog ~/datasets/cat_dog Hi @halimatif , it looks like you installed PyTorch for Python 3.6, but are attempting to run the script with Python 2.7. If you change python to python3 when launching the script above, does it work?

Yes, it works! Thank you. Rookie mistake… I was literally following the exercise “Re-training ResNet-18 Model” listed steps without understanding the connection between “PyTorch for Python 3.6” and the “python” reference in the launch training command.

Thanks, glad you were able to get it running, and I will have to make a note of that in the documentation. More than likely I will move all the references over to python3 since Python 2.7 is EOL.

Hello AI World on Jetpack 4.4

Robotics & Edge Computing Jetson Systems Jetson Nano

dusty_nv May 19, 2020, 4:25pm 5

I’m currently fixing the install_pytorch.sh script to detect JetPack 4.4 and install the different pip wheels, sorry about that. It should be checked in later today. For now, you can install PyTorch manually from here: PyTorch for Jetson

Topic		Replies	Views
Pytorch on Jetson nano Jetpack 4.4 R32.4.4 Jetson Nano pytorch	3	1581	November 17, 2020
Issues installing PyTorch on Jetson Nano running JetPack 4.2 Frameworks (archived) pytorch	3	2604	June 5, 2020
Problem with the python train.py command from "Hello AI World" Jetson TX2 ai-training	2	618	June 23, 2020
Failed install pytorch in jetson nano Jetson Nano jetson-inference , pytorch	8	758	December 22, 2023
Torchvision will not import into Python after jetson-inference build of PyTorch Jetson Nano jetson-inference , pytorch , python	14	2329	July 11, 2020
Python3 train_ssd.py --data=data/fruit --model-dir=models/fruit --batch-size=4 --epochs=30 Jetson Nano pytorch	27	2868	October 27, 2021
Hello AI World Training Cat/Dog Jetson Nano ai-training	6	671	March 29, 2024
Jet pack version for "jetson-nano-jp461-sd-card-image"? Jetson Nano pytorch	3	448	April 23, 2024
Issues Installing pytorch through jetson inference install script Jetson Nano pytorch	6	1512	October 2, 2023
Enable to import torch and torch-vision after installing it with Jetson Inference github repository Jetson Nano kernel , pytorch	2	119	October 20, 2025

Hello AI World on Jetpack 4.4

Related topics