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

5

Hello,
Thank you for your response. I am using Jetson Orin Nano Developer kit with Jetpack 6.0.
Libraries I am using are as follows:
CUDA: 12. 2.140
cuDNN: 1.0
TensorRT: 8.6.2.3
VPI: 3.1.5
Vulkan: 1.3.204
openCV: 4.8.0 with CUDA: No
Python: 3.10.204

I will try to follow the article you suggested. Hope it will help.

Thanks again.

1 Like