search

grimoire / amirstan_plugin

Useful tensorrt plugin. For pytorch and mmdetection model conversion.
MIT License
159 stars 38 forks source link

Error at /usr/share/cmake-3.10/Modules/FindCUDA.cmake:682 #17

Open BChunlei opened 3 years ago

BChunlei commented 3 years ago

when i install amirstan_plugin and use the command 'cmake -DTENSORRT_DIR=/root/downloads/TensorRT-7.2.3.4/ ..',error occurs:

-- The C compiler identification is GNU 7.5.0 -- The CXX compiler identification is GNU 7.5.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 CMake Error at /usr/share/cmake-3.10/Modules/FindCUDA.cmake:682 (message): Specify CUDA_TOOLKIT_ROOT_DIR Call Stack (most recent call first): CMakeLists.txt:6 (FIND_PACKAGE)

-- Configuring incomplete, errors occurred! See also "/root/project/amirstan_plugin/build/CMakeFiles/CMakeOutput.log".