Pytorch on gpu

Robotics & Edge Computing Jetson Systems Jetson Orin NX
1

I am using JetPack 6.2.1 on a Jetson Orin NX 8GB, but there are no official PyTorch GPU libraries available for this version. I tried loading my model using a TensorRT engine, but it still requires PyTorch GPU support. Please help me resolve this issue.

5

Hi,

Please refer to jp6/cu126 index to get the prebuilt wheels

Thanks

6

Yes, it was downloaded successfully, but while running my YOLO models, I kept getting the following errors:
CRASH: CUDA error: out of memory,
NvMapMemHandleAlloc: error 0, and
RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED.

7

Hi,

Seems like out of memeory, please try the following jetson-containers/docs/setup.md at master · dusty-nv/jetson-containers · GitHub suggetions.

Thanks