libtorch基本使用
在pytorch官网下载C++版本的libtorch(这是编译好的)。一般下载release版本就行了。解压。如果下载的是Windows版本,需要提前装好vs2017或以上的版本(2015会报错)我习惯使用cmake进行编译。(vs2019可以创建cmake工程)如果你用的是vs2017,那么可以安装一个clion。编译cmake程序非常方便。配置选x64平台(很重要,否则报错),选rele...
在pytorch官网下载C++版本的libtorch(这是编译好的)。一般下载release版本就行了。解压。
如果下载的是Windows版本,需要提前装好vs2017或以上的版本(2015会报错)
我习惯使用cmake进行编译。(vs2019可以创建cmake工程)如果你用的是vs2017,
那么可以安装一个clion。编译cmake程序非常方便。配置选x64平台(很重要,否则报错),选release。即可
把cmake文件改为:(这里只需要修改CMAKE_PREFIX_PATH为你自己的路径即可)
cmakelist.txt:
cmake_minimum_required(VERSION 3.0 FATAL_ERROR)
project(example-app)
set(CMAKE_PREFIX_PATH D:/libtorch)
find_package(Torch REQUIRED)
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${TORCH_CXX_FLAGS}")
add_executable(example-app main.cpp)
target_link_libraries(example-app "${TORCH_LIBRARIES}")
set_property(TARGET example-app PROPERTY CXX_STANDARD 11)
# The following code block is suggested to be used on Windows.
# According to https://github.com/pytorch/pytorch/issues/25457,
# the DLLs need to be copied to avoid memory errors.
if (MSVC)
file(GLOB TORCH_DLLS "${TORCH_INSTALL_PREFIX}/lib/*.dll")
add_custom_command(TARGET example-app
POST_BUILD
COMMAND ${CMAKE_COMMAND} -E copy_if_different
${TORCH_DLLS}
$<TARGET_FILE_DIR:example-app>)
endif (MSVC)
main.cpp:
#include <iostream>
#include "torch/torch.h"
#include "torch/jit.h"
int main() {
std::cout << "Hello, World!" << std::endl;
auto a = torch::tensor({{1, 2}, {3, 4}});
std::cout << a << std::endl;
return 0;
}
编译,即可体验C++版本的pytorch。libtorch调用模型有两种方式,官方给的例子很清楚:
更多推荐
所有评论(0)