Ubuntu/Debian pip install kaldifeat: cannot find CUDNN

[pzelasko]

either doing pip install kaldifeat, or cloning source and doing python setup.py install, yields the following message:

...
-- Found CUDA: /usr/local/cuda (found version "10.2")
-- Caffe2: CUDA detected: 10.2
-- Caffe2: CUDA nvcc is: /usr/local/cuda/bin/nvcc
-- Caffe2: CUDA toolkit directory: /usr/local/cuda
-- Caffe2: Header version is: 10.2
-- Could NOT find CUDNN (missing: CUDNN_LIBRARY_PATH CUDNN_INCLUDE_PATH)
CMake Warning at /home/pzelasko/miniconda3/envs/lhotse/lib/python3.7/site-packages/torch/share/cmake/Caffe2/public/cuda.cmake:109 (message):
  Caffe2: Cannot find cuDNN library.  Turning the option off
Call Stack (most recent call first):
  /home/pzelasko/miniconda3/envs/lhotse/lib/python3.7/site-packages/torch/share/cmake/Caffe2/Caffe2Config.cmake:88 (include)
  /home/pzelasko/miniconda3/envs/lhotse/lib/python3.7/site-packages/torch/share/cmake/Torch/TorchConfig.cmake:68 (find_package)
  cmake/torch.cmake:13 (find_package)
  CMakeLists.txt:41 (include)

I am able to build it by doing this:

<inside sources root>
mkdir build
cd build
cmake .. -DCUDNN_LIBRARY_PATH="/home/pzelasko/miniconda3/envs/lhotse/lib" -DCUDNN_INCLUDE_PATH="/home/pzelasko/miniconda3/envs/lhotse/include"

But there seems to be no way to pass these values through env vars when doing pip install kaldifeat.

[csukuangfj]

But there seems to be no way to pass these values through env vars when doing pip install kaldifeat.

There are some ways to pass these env vars to it, either using pip install or using python3 setup.py install.

Will propose a fix.

[csukuangfj]

I just realized the current master supports that.

You need to do

$ export KALDIFEAT_CMAKE_ARGS='-DCMAKE_BUILD_TYPE=Release -DCUDNN_LIBRARY_PATH="/home/pzelasko/miniconda3/envs/lhotse/lib" -DCUDNN_INCLUDE_PATH="/home/pzelasko/miniconda3/envs/lhotse/include"'

$ echo $KALDIFEAT_CMAKE_ARGS
-DCMAKE_BUILD_TYPE=Release -DCUDNN_LIBRARY_PATH="/home/pzelasko/miniconda3/envs/lhotse/lib" -DCUDNN_INCLUDE_PATH="/home/pzelasko/miniconda3/envs/lhotse/include"

And then you can run either python3 setup.py install or pip install kaldifeat.
It should use the CMake args you specified in KALDIFEAT_CMAKE_ARGS.

---

EDITED:

It should be

-DCUDNN_LIBRARY_PATH="/home/pzelasko/miniconda3/envs/lhotse/lib/libcudnn.so"

See k2-fsa/icefall#120 (comment)

[csukuangfj]

The code is in

kaldifeat/cmake/cmake_extension.py

Line 59 in 3de5b99

cmake_args = os.environ.get("KALDIFEAT_CMAKE_ARGS", "")

kaldifeat/cmake/cmake_extension.py

Lines 63 to 64 in 3de5b99

	if cmake_args == "":
	cmake_args = "-DCMAKE_BUILD_TYPE=Release"

kaldifeat/cmake/cmake_extension.py

Line 77 in 3de5b99

cmake {cmake_args} {kaldifeat_dir}

[WangHelin1997]

The same error and solved with it. Thanks!