cuDNN on DRIVE Software?

Autonomous Vehicles DRIVE AGX Xavier DRIVE AGX Xavier General
1

Hardware Platform: DRIVE AGX Xavier™ Developer Kit
Software Version: DRIVE Software 10
Host Machine Version: native Ubuntu 18.04
SDK Manager Version: Example: 1.3.0.7105

Hello, I’m currently trying to run a TensorFlow (1.15.4) program on the AGX but it can’t find cuDNN and runs on the CPU. Any possibilities cuDNN can be installed on the AGX for GPU support?
The install guide only refers to amd64 deb files whereas I need the arm64 version.

Thanks

2

Dear @Spoekels,
Cudnn on DRIVE platform is part of DRIVE SW. There is no separate release for DRIVE platform.
Could you please check Installing Tensorflow from source on DRIVE AGX if it helps.

Also, we recommend using TensorRT on DRIVE AGX for good performance. You may convert your TF model to TensorRT and perform inference.