Torch version for Jetson Nano, Jetpack 6.0 and how to install it?

Robotics & Edge Computing Jetson & Embedded Systems Jetson Xavier NX
1

Hello,
I recently started working on jetson nano. While I am running this command on the jetson nano,

$flresearch@flresearch-desktop:~/yolov5$ python3 train.py --img 640 --batch 4 --epochs 2 --data data/VOS2028.yaml --weights ‘’ --cfg models/yolov5s.yaml --device 0

I am hitting with one error,
Traceback (most recent call last):
File “/home/flresearch/yolov5/train.py”, line 34, in
import torch
ModuleNotFoundError: No module named ‘torch’

I am using jetpack 6.0, I would really appreciate if you could tell me which version of torch ( I search online and found that the most suitable for jetpack 6.0 will be pytorchv2.2.0) I need to install and where can i get those instruction?
Thanks for your help in advance.

~Monika

4

Hi,

Is your device a Jetson Nano or Xavier NX?

Seems you are using JP6.0, which not compatible with Jetson Nano and Xavier NX.
Is your device a Orin nano?

We provide prebuilt wheel in

Please refer it.

Thanks