NVIDIA Developer Forums
What is the compatible torchvision version for torch-1.14.0a0+44dac51c.nv23.2 installed on AGX ORIN( Ubuntu-22.04, JP-5.1/5.0, Python-3.8, CUDA-11.4)
Autonomous Machines
Jetson & Embedded Systems
Jetson AGX Orin
pytorch
fpsychosis
April 18, 2023, 2:38pm
14
'same with 0.14.1, gpu not recognized
Screenshot from 2023-04-18 15-34-21
1280×720 171 KB
@dusty_nv
I could read in your docker you used 0.14.1 with torch 2.0, how you avoided this error and compiled with GPU?
Torchvision on Jetson AGX Orin DevKit
show post in topic