I found a lot of threads of issues with PX2 and Tensorflow and Docker, however I couldnt get a comprehensive impression of what is the current status. https://devtalk.nvidia.com/default/topic/1049100/general/tensorflow-installation-on-drive-px2-/
Could you please clarify:
If I use a tensorflow-gpu docker from the official tensorhub, will I be able to use the gpus on the nvidia?
Is there any other custom dockerfile that enables the gpus support inside a docker?
If I cant use docker, is there a prebuilt version of tensorflow with python3 for the px2?
Does the prebuilt version with python 2.7 support the gpus natively?
The ideal solution would be the use of a docker similar to the way to use it on a regular pc with nvdia-docker for the tensorflow needs. Does this possibility exist?
You will need to use nvidia-docker to have a GPU version TensorFlow.
However, nvidia-docker doesn’t support PX2.
A possible solution is to install TensorFlow directly on the PX2 without using docker.
This will require you to build it from source for the aarch64 support.
You can check this topic for the building steps. Or install this prebuilt file directly.
The prebuilt version did not install on our PX2, I posted this in the thread you linked. But if docker is not an option we will have another look at it.