Jetson 6.1 running ultralytics

[jakesheehan101]

new to all this so bare with me.
I’m running
Jetpack 6.1
cuda 12.6
cuDNN 9.3
pytorch 2.5.1
numpy 1.23.5

cuda.is_available will run true in pycharm and console, but when I import ultralytics from pycham package manager it stops running.
Ive tried installing ultralytics from console, no luck
tried upgrading numpy from consoe but it breaks cuda
upgraded numpy 2.1.3 and torch 2.5.1 no luck

thanks for the help

[jakesheehan101]

for anyone coming after I ended up tracing it down to torchvision version issues. downgraded to 0.12.0 and it started working.