I’m working with Jetson AGX Xavier and I’m trying to get tensorflow 2 working on it.
I’ve found a working wheel for tensorflow 2 here: https://github.com/lhelontra/tensorflow-on-arm/releases/
but while testing it I’ve discovered it doesn’t have gpu support.
I’ve also checked docker images from nvidia: https://ngc.nvidia.com/catalog
but still nothing for arm64 with tensorflow 2.
Does anyone knows where I can get a docker or wheel with tensorflow 2 for arm64 to work on the Jetson?