ncnn 编译与使用 pnnx 编译与使用

AI_潜行者

已于 2022-07-18 03:43:15 修改

阅读量1.1k

点赞数

CC 4.0 BY-SA版权

分类专栏： C++ 环境配置源码编译文章标签： git github ncnn

于 2022-07-15 10:32:46 首次发布

本文链接：https://blog.youkuaiyun.com/weixin_40011280/article/details/125796741

环境配置同时被 3 个专栏收录

20 篇文章

订阅专栏

C++

9 篇文章

订阅专栏

源码编译

5 篇文章

订阅专栏

本文档详细介绍了如何编译和使用NCNN，一个由腾讯开发的高性能神经网络前向计算框架。首先，从GitHub克隆NCNN项目到本地，然后使用Visual Studio进行编译。此外，还阐述了PNNX的配置步骤，特别是针对libtorch的设置，包括CMAKE_PREFIX_PATH的配置以及解决DLL定义冲突的问题。最后，指导读者在完成配置后，通过Visual Studio生成项目。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

ncnn 编译与使用 pnnx 编译与使用

ncnn

1.ncnn的官网：https://github.com/Tencent/ncnn

2.clone到本地：git clone git@github.com:Tencent/ncnn.git

3.visual studio打开：右键全部生成

pnnx

1.位置：pnnx位于ncnn的 toos/pnnx

2.配置 libtorch ：

使用 CMAKE_PREFIX_PATH 配置：
首先找到 TorchConfig.cmake 的目录（可用everything工具），然后list(append CMAKE_PREFIX_PATH …)

if(CMAKE_BUILD_TYPE AND (CMAKE_BUILD_TYPE STREQUAL "Debug"))
    list(APPEND CMAKE_PREFIX_PATH "D:/lib/libtorch-win-shared-with-deps-debug-1.9.0+cu111/libtorch/share/cmake")
elseif(CMAKE_BUILD_TYPE AND (CMAKE_BUILD_TYPE STREQUAL "Release"))
    list(APPEND CMAKE_PREFIX_PATH "D:/lib/libtorch-win-shared-with-deps-1.9.0+cu111/libtorch/share/cmake")
endif()

注意：如果遇上 torch_cpu.lib(torch_cpu.dll) 和 torch.lib(torch.dll) 中定义多重定义的 dll，则取消链接 torch.lib即可，我们需要去 TorchConfig.cmake 找到存库名的变量 TORCH_LIBRARIES

# 找到TORCH_LIBRARIES后，首获查看得所有变量全路径
foreach( __val ${TORCH_LIBRARIES} )
	message(${__val})
endforeach()

# 直接删除
list(REMOVE_ITEM TORCH_LIBRARIES "-WHOLEARCHIVE:D:/lib/libtorch-win-shared-with-deps-debug-1.9.0+cu111/libtorch/lib/torch.lib")
# 或者这总方式（可以采用正则）
foreach( __val ${TORCH_LIBRARIES} )
    string(COMPARE EQUAL ${__val} "-WHOLEARCHIVE:D:/lib/libtorch-win-shared-with-deps-debug-1.9.0+cu111/libtorch/lib/torch.lib" FLAG)
    if(FLAG)
        list(REMOVE_ITEM  TORCH_LIBRARIES  ${__val})
    endif()
endforeach()