JAX wheels for Jetson Nano / JetPack 4.6

Robotics & Edge Computing Jetson & Embedded Systems Jetson Nano
1

JAX doesn’t distribute binaries compatible with the Jetson Nano. Building it from source is a chore. Here are some wheels which are compatible with the system Python 3.6 / CUDA 10.2 installation. (You may need to upgrade NumPy and SciPy).

3

Thanks for the sharing : )

5

This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.