Installation of TensorFlow on Drive PX

Is there any update on this?
As the latest version of DriveInstall ships with CUDA 9.2 the wheels are not working any more!

How can I install tensorflow on the DPX2 with CUDA 9.2?


There is no prebuilt TensorFlow package for CUDA 9.2.
Please build it from the source.

Here is the tutorial for your reference:

Please update the GPU architecture for DRIVE PX2.

Please specify a list of comma-separated Cuda compute capabilities you want to build with. 
You can find the compute capability of your device at: 
Please note that each additional compute capability significantly increases your build time and binary size. [Default is: 3.5,5.2] 



I’ve followed the instructions and build tensorflow 1.12 for the PX2. The GPU is detected by keras/tf.
However if we start inference, the following error is show:

Error : Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above

For completion we are using the following versions:

  • python: 3.5
  • cuda 9.2 at /usr/local/cuda/
  • cuDNN 7.1.2 at /usr/lib/aarch64-linux-gnu
  • NCCL 1.3 (automatically fetched)
  • Is the NCCL version right?


Here is a prebuilt package and some installation steps for your reference: