MxNet support for Jetson Orin modules (Cuda 11.4)

Check failed: err == cudaSuccess (209 vs. 0) : mxnet_generic_kernel ErrStr:no kernel image is available for execution on the device

Has anyone been able to install mxnet on Jetson Orin successfully? I tried 1.9.1 and it runs into cuda error mentioned above when testing the final GPU usage step in the instructions.

Jetson Orin comes with cuda 11.4 only and I couldn’t downgrade it to the supported cudas 11.2 or below. Any help is much appreciated.

thanks,
Anil

It works now, I had to modify the arch from 72 to 87 for Orin modules -gencode arch=compute_72,code=sm_72 for Xavier NX and AGX Xavier in the config make file