You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
{{ message }}
This repository was archived by the owner on Nov 17, 2023. It is now read-only.
Repository navigation
This repository was archived by the owner on Nov 17, 2023. It is now read-only.
mxnet 1.9.1 cuda support for Jetson AGX Orin (Jetpack 5.0 with cuda 11.4) #21166
Welcome to Apache MXNet (incubating)! We are on a mission to democratize AI, and we are glad that you are contributing to it by opening this issue.
Please make sure to include all the relevant context, and one of the @apache/mxnet-committers will be here shortly.
If you are interested in contributing to our project, let us know! Also, be sure to check out our guide on contributing to MXNet and our development guides wiki.
changed the title [-]mxnet 1.9.1 cuda error with cuda 11.4 on Jetson AGX Orin (Jetpack 5.0)[/-][+]mxnet 1.9.1 cuda support for Jetson AGX Orin (Jetpack 5.0 with cuda 11.4)[/+]on Jan 9, 2023
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
Description
mxnet (1.9.1) is raising a cuda error with cuda 11.4 on Jetson AGX Orin series with Jetpack 5.0
Error Message
Check failed: err == cudaSuccess (209 vs. 0) : mxnet_generic_kernel ErrStr:no kernel image is available for execution on the device
To Reproduce
I ran the installation steps recommended for the jetson and built things manually from source with cuda path and other variables.
Everything went fine with the make and importing mxnet and version check. When I run the gpu check the error pops up.
What have you tried to solve it?
Environment
Ubuntu 20.04, Nvidia jetson AGX Orin developer kit
Environment Information