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后虽然还有警告,但是编译可以通过,提供的测试程序也可以正常运行。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值