[ENHANCEMENT] Let cmake find cudnn libraries from /usr/lib/x86_64-linux-gnu and /usr/include

microsoft / nnfusion

A flexible and efficient deep neural network (DNN) compiler that generates high-performance executable from a DNN model description.

MIT License

952 stars 158 forks source link

[ENHANCEMENT] Let cmake find cudnn libraries from /usr/lib/x86_64-linux-gnu and /usr/include #376

Open xysmlx opened 2 years ago

xysmlx commented 2 years ago

🚀 Feature

Let cmake find cudnn libraries from /usr/lib/x86_64-gnu and /usr/include.

Motivation

NVIDIA official docker image installs cudnn (i.e., libcudnn[version]-dev) in /usr/lib/x86_64-linux-gnu (for runtime files) and /usr/include (for headers). Current nnfusion's cmake config cannot find cudnn because it is not installed in /usr/local/cuda.

Pitch

Alternatives

Additional context