Skip to content

added cudnn + tensorrt

Maxandre Ogeret requested to merge MaxandreOgeret/docker:cudnn_tensorrt into master

Implemented feature

Make cuDNN and Tensorrt available as some features needs it, like pointpillars for example.

Implementation considerations

Using the nvidia machine learning learning repo and apt-get

Alternatives

n/a

Additional information

n/a

Implementation description

The Nvidia machine learning repo was already available so I just had to identify the right version of the package and install them using apt-get.

Pointpillars compiles fine now.

Merge request reports