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()