libtorch环境配置

环境配置

建议在linux上配置对应环境

可以在autoDL上租一个服务器来搭建，带有pytorch的环境
https://www.autodl.com/home
我自己的win电脑上安装了pytorch，但是配置时会报错，于是到ubuntu上配置

电脑上装有pytorch的就不需要再下载libtorch了，pytorch就带有libtorch
首先通过下面的代码找出pytorch自带的环境

import torch
torch.utils.cmake_prefix_path

整体架构图

编写CMakeLists.txt文件

cmake_minimum_required(VERSION 3.6 FATAL_ERROR)
project(test-libtorch)
# 设置libtorch的位置
set(CMAKE_PREFIX_PATH "/root/miniconda3/lib/python3.8/site-packages/torch/share/cmake")
message(${CMAKE_PREFIX_PATH})
find_package(Torch REQUIRED)
#op.cpp exe
add_executable(test-libtorch op.cpp)

# #link libtorch .a .so
target_link_libraries(test-libtorch "${TORCH_LIBRARIES}")
#
set_property(TARGET test-libtorch PROPERTY CXX_STANDARD 14)

编写op.cpp

#include<torch/torch.h>
#include <iostream>

int main() {
   
   
	torch::Tensor tensor = torch::rand({
   
    2, 3 });
	std::cout << tensor << std::endl;
	return 0;
}

编译

先创建build文件夹，再在build里面编译