TensorRT-CenterNet
TensorRT-CenterNet copied to clipboard
/usr/bin/ld: cannot find -lcudart /usr/bin/ld: cannot find -lcublas
This is not a issue. When I compile the onnx-tensorrt folder
ubuntu 16.04 cmake 3.5 cuda 10.0 cudnn 7.5
It seems that there is a problem in CMakeLists.txt.
/usr/bin/ld: cannot find -lcudart /usr/bin/ld: cannot find -lcublas collect2: error: ld returned 1 exit status ...
I solved it from https://devtalk.nvidia.com/default/topic/487573/linker-error-usr-bin-ld-cannot-find-lcudart/ by replace cuda cudart cublas
into ${CUDA_LIBRARIES}
in lines
https://github.com/CaoWGG/TensorRT-CenterNet/blob/f949252e37b51e60f873808f46d3683f15735e79/onnx-tensorrt/CMakeLists.txt#L244
In TensorRT-Centernet/CMakeLists.txt add LINK_DIRECTORIES
solved for me !