I’ve tried using the wheels linked below for getting Tensorflow on the PX2. They worked pretty alright for the Parker GPGPU and the CPU. However, I could not get it to work with the Pascal GPU as the wheels were built with the CUDA compute capability set to 6.2 (which is correct for the TX2 GPU architecture). But the Pascal GPU on the PX2 has a CUDA compute capability of 6.1. I am therefore wondering if you have any wheels built with a compute capability of 6.1, or if you could build new versions of these wheels with the right parameters for the PX2 GPU architecture.
I’ve also tried building Tensorflow from source, however I ran into a lot of issues doing this.