JAX on Orin

Has anyone had success with installing JAX with GPU support on the Orin? I have followed steps for other Jetson devices from this thread, but have had no luck. Building from source with and without cuda results in the error messages attached.

Any help would be greatly appreciated

jax_build_log_cuda.txt (13.0 KB)
jax_build_log_cpu.txt (12.2 KB)

There is no update from you for a period, assuming this is not an issue any more.
Hence we are closing this topic. If need further support, please open a new one.


Based on the below error:

ERROR: ... Compiling tensorflow/compiler/xla/service/cpu/runtime_fp16.cc failed: (Exit 1): crosstool_wrapper_driver_is_not_gcc failed: error executing command 

It seems there are some issues in the bazel setting.
Could you share the configuration you used with us as well?