search

HawkAaron / warp-transducer

A fast parallel implementation of RNN Transducer.
Apache License 2.0
307 stars 124 forks source link

cuda_curand_library not found at build time #85

Closed gandroz closed 3 years ago

gandroz commented 3 years ago

I use the official tensorflow docker image. I want to install RNNT wrapper to TF but it fails finding some libraries

cmake         -DUSE_NAIVE_KERNEL=on         -DCMAKE_C_COMPILER_LAUNCHER="$(which gcc)"         -DCMAKE_CXX_COMPILER_LAUNCHER="$(which g++)"          -DCUDA_TOOLKIT_ROOT_DIR=$CUDA_HOME ..
-- The C compiler identification is GNU 8.4.0
-- The CXX compiler identification is GNU 8.4.0
-- Check for working C compiler: /usr/bin/cc
-- Check for working C compiler: /usr/bin/cc -- works
-- Detecting C compiler ABI info
-- Detecting C compiler ABI info - done
-- Detecting C compile features
-- Detecting C compile features - done
-- Check for working CXX compiler: /usr/bin/c++
-- Check for working CXX compiler: /usr/bin/c++ -- works
-- Detecting CXX compiler ABI info
-- Detecting CXX compiler ABI info - done
-- Detecting CXX compile features
-- Detecting CXX compile features - done
-- Looking for pthread.h
-- Looking for pthread.h - found
-- Looking for pthread_create
-- Looking for pthread_create - not found
-- Looking for pthread_create in pthreads
-- Looking for pthread_create in pthreads - not found
-- Looking for pthread_create in pthread
-- Looking for pthread_create in pthread - found
-- Found Threads: TRUE  
-- Found CUDA: /usr/local/cuda (found version "11.0") 
-- cuda found TRUE
-- Building shared library with GPU support
-- CuRand library CUDA_curand_LIBRARY-NOTFOUND
CMake Error: The following variables are used in this project, but they are set to NOTFOUND.
Please set them or make sure they are set and tested correctly in the CMake files:
CUDA_curand_LIBRARY (ADVANCED)
    linked by target "test_time_gpu" in directory /src/warp-transducer
    linked by target "test_gpu" in directory /src/warp-transducer

and that's the content of the cuda libraries

libOpenCL.so             libcublasLt.so.11.2.0.252  libcuinj64.so            libcusolver.so.10            libnvblas.so.11.2.0.252
libOpenCL.so.1           libcudadevrt.a             libcuinj64.so.11.0       libcusolver.so.10.6.0.245    libnvperf_host.so
libOpenCL.so.1.0         libcudart.so               libcuinj64.so.11.0.221   libcusolverMg.so.10          libnvperf_host_static.a
libOpenCL.so.1.0.0       libcudart.so.11.0          libculibos.a             libcusolverMg.so.10.6.0.245  libnvperf_target.so
libaccinj64.so           libcudart.so.11.0.221      libcupti.so              libcusparse.so.11            libnvrtc-builtins.so
libaccinj64.so.11.0      libcudart_static.a         libcupti.so.11.0         libcusparse.so.11.1.1.245    libnvrtc-builtins.so.11.0
libaccinj64.so.11.0.221  libcufft.so.10             libcupti.so.2020.1.1     libnvToolsExt.so             libnvrtc-builtins.so.11.0.221
libcublas.so.11          libcufft.so.10.2.1.245     libcupti_static.a        libnvToolsExt.so.1           libnvrtc.so.11.0
libcublas.so.11.2.0.252  libcufftw.so.10            libcurand.so.10          libnvToolsExt.so.1.0.0       libnvrtc.so.11.0.221
libcublasLt.so.11        libcufftw.so.10.2.1.245    libcurand.so.10.2.1.245  libnvblas.so.11              stubs

It seems that cuda_curand (at least!) is not found.

Also, I noticed this line in the CMakeLists file, but I dont see where Torch is eventually searched ?