libtorch-yolov3安装记录

环境:
cuda 11.1
cudnn 8.0
libtorch 1.7

本noob被折磨了两天,出现的问题有两个

  1. cmake生成文件的时候:
    Caffe2:Header file 里的cuda版本显示9.0,nvcc的是11.1,报错
  2. 如果第一个问题没出现,就会报警告,无法生成安全的运行时版本??
    can not generate safe runtime…
    cuda的库和/usr/lib/x86… 下的一个库覆盖了?
    这个时候再make就会报错,后面发现可能和c++的版本有关
    上面两个问题参照pytorch官方教程的CMakelist对原本的cmakelist做了修改
cmake_minimum_required(VERSION 3.0 FATAL_ERROR)
project(yolo-app)
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS}  ${TORCH_CXX_FLAGS}")#和官方教程一致

find_package( OpenCV REQUIRED )
include_directories( ${OpenCV_INCLUDE_DIRS} ${Torch_DIR})#修改

set(Torch_DIR /home/zsp/ThirdPartyLib/libtorch/share/cmake/Torch)  # 我的 libtorch 的路径
find_package(Torch REQUIRED)

aux_source_directory(. DIR_SRCS)

add_executable(yolo-app ${DIR_SRCS})
target_link_libraries(yolo-app ${TORCH_LIBRARIES} ${OpenCV_LIBS})
set_property(TARGET yolo-app PROPERTY CXX_STANDARD 14)#设置c++标准为c14

关键可能在c++标准?设置成c14后虽然还有警告,但是编译可以通过,提供的测试程序也可以正常运行。

你可能感兴趣的:(本noob被折磨记录)