cmake error

[ChanghaoZheng]

Env

• GPU, NVIDIA GeForce GTX 1650
• OS, Ubuntu18.04
• Cuda version:None
• TensorRT version:yolov5-v5.0

About this repo

• which branch/tag/commit are you using? tensorrtx-yolov5-v5.0
• which model? yolov5, retinaface? yolov5-5.0,yolov5s.pt

Your problem

• what is your command? 'cmake ..'
• what's your output?-- 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:11 (find_package)

-- Configuring incomplete, errors occurred!
See also "/home/zheng/yolov5-deepsort-tensorrt-c++/tensorrtx-yolov5-v5.0/yolov5/build/CMakeFiles/CMakeOutput.log".

• what output do you expect?
  successly building

[wang-xinyu]

CUDA is required.