Expecting torch and torchvision for JP6.1

Support YOLO
, , ,
7

I came across the same problem with Jetson AGX / Jetpack 6.1

I used torch from Dustin’s Site.

wget http://jetson.webredirect.org/jp6/cu126/+f/5cf/9ed17e35cb752/torch-2.5.0-cp310-cp310-linux_aarch64.whl#sha256=5cf9ed17e35cb7523812aeda9e7d6353c437048c5a6df1dc6617650333049092

pip install torch-2.5.0-cp310-cp310-linux_aarch64.whl

then installed them respective torchvision from Dustin’s site.

wget http://jetson.webredirect.org/jp6/cu126/+f/5f9/67f920de3953f/torchvision-0.20.0-cp310-cp310-linux_aarch64.whl#sha256=5f967f920de3953f2a39d95154b1feffd5ccc06b4589e51540dc070021a9adb9

pip install torchvision-0.20.0-cp310-cp310-linux_aarch64.wh

then installed onnxruntime 1.20 from justin s site.

wget http://jetson.webredirect.org/jp6/cu126/+f/0c4/18beb3326027d/onnxruntime_gpu-1.20.0-cp310-cp310-linux_aarch64.whl#sha256=0c418beb3326027d83acc283372ae42ebe9df12f71c3a8c2e9743a4e323443a4

pip install onnxruntime_gpu-1.20.0-cp310-cp310-linux_aarch64.whl

Downgraded the numpy as usual.

and lastly installed libcusparselt0 from nvidia site.

1 Like