第9节：libtorch开发深度学习算法中的autograde

AI拉呱

于 2022-08-05 22:41:50 发布

阅读量477

点赞数

分类专栏： C++复现机器学习和深度学习算法文章标签：深度学习算法计算机视觉

本文链接：https://blog.youkuaiyun.com/weixin_32393347/article/details/126186900

版权

C++复现机器学习和深度学习算法同时被 2 个专栏收录

23 篇文章 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

C/C++编程

33 篇文章

订阅专栏

本篇博客介绍如何在C++中利用libtorch进行深度学习算法开发，通过CmakeLists.txt配置文件来组织项目。

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

CmakeLists.txt

cmake_minimum_required (VERSION 3.8)

project(SOLDIER)
set(Torch_DIR "/libtorch/share/cmake/Torch")
set(PYTHON_EXECUTABLE "/usr/bin/python3")


find_package(Torch REQUIRED)
find_package(OpenCV REQUIRED)

set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS}")


set(CMAKE_BUILD_TYPE Debug)

include_directories(${CMAKE_SOURCE_DIR}/include)
include_directories(${OpenCV_INCLUDE_DIRS})

add_executable(run main.cpp)
target_link_libraries(run "${OpenCV_LIBS}" "${TORCH_LIBRARIES}")
set_property(TARGET run PROPERTY CXX_STANDARD 14)

C++


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

了解本专栏

超级会员免费看