玩深度学习,个人觉得基于anaconda的python适合开发与测试,C++适合实际的工程部署!而pytorch官方有编译好的libtorch,特别方便,适合于我这样的伸手党和手残党(win10下编译tensorflow编译了好久都没通过,好忧伤!)好了,话不多说进入正题!
工程环境:
1.win10;
2.anaconda+pytorch1.0;
3.vs2015
4.Cmake
一、下载libtorch
进入pytorch官方网址https://pytorch.org/get-started/locally/,个人使用的是cpu版本,选择如下图所示:
下载好了放在无中文路径的文件夹下(个人使用的是release版本)
二、Cmake配置
首先创建一个文件夹,存放主程序example.cpp
还有CMakeLists.txt
,然后再创建一个build的空文件夹,之后编译好的文件都存放在build文件夹里头(照着教程来的,但是发现build文件夹内实际没有工程文件,反而是根目录文件夹内是编译好的文件,即CMakeLists.txt所在文件夹,有知道的大佬烦请告知下原因
)。cmake编译文件夹目录如下所示:
其中CMakeLists.txt内容如下所示:
#cmake的最低版本要求
cmake_minimum_required(VERSION 3.0 FATAL_ERROR)
#example为创建的工程名字
project(example)
#添加的pytorch的附加库
find_package(Torch REQUIRED)
#编译的相应cpp文件
add_executable(example-app example-app.cpp)
target_link_libraries(example-app “${TORCH_LIBRARIES}”)
set_property(TARGET example-app PROPERTY CXX_STANDARD 11)
其中example.cpp内容如下所示(仅作调用libtorch并生成随机tensor):
#include <torch/torch.h>
#include <iostream>int main() {torch::Tensor tensor = torch::rand({5, 3});std::cout << tensor << std::endl;system("pause");
}
三、Cmake编译
camke编译比较简单,在根目录文件夹下打开power shell(即CMakeLists.txt所在文件夹
),输入
cmake -DCMAKE_PREFIX_PATH=E:\lib\pytorch\libtorch
DCMAKE_PREFIX_PATH=libtorch绝对路径,就是刚才下载的libtorch文件解压后的根目录。
编译成功后会出现以下结果(并且刚才相应的文件会出现VS工程项目):
四、结果运行
用vs2015打开如下文件
1.打开工程文件后,修改配置管理器为X64,以及将example项目设置为启动项目(如果不会,请自行百度,这是vs的基本操作)。2.打开example项目的属性->链接器->输入->附加依赖项,删除其中引号内的所有内容,否则会出现找不到torch.lib的错误提示;
3.接上述步骤,->命令行->其他选项,将其中的X86更改为X64;
运行example项目会提示找不到相应的lib文件,此时继续下面步骤,
4.将之前解压的libtorch文件(\path\libtorch\lib)下的lib文件夹内的所有内容(内容如下图所示)拷贝至\path\example-app\x64\Release文件夹下,
此时项目运行成功,结果如下图所示:
注意:
1.CamkeLists.txt,主要为设置相应的项目属性以及需要编译的附加库,但是不明白为什么还需要将相应的libtorch库文件拷贝至当前项目环境下,难道是因为没有在环境变量里设置libtorch的变量路径吗?下次有空再测试;
2.libtorch解压路径不要太深,貌似是因为Cmake路径不能超过一千个字符,所以路径尽量浅一点,还有就是作为码农,不要有中文路径,毕竟外国人基本不用中文,在码农世界中文是个蛋疼的玩意,会导致很多莫名其妙的错误,避免坑就戒掉中文路径吧!
3.Camke是个好东西,可以多学习学习,下面有相应的CamkeLists.txt的参考博客,觉得都是大牛啊!
参考文献:
1.https://blog.csdn.net/adyf3/article/details/85003893
2.https://blog.csdn.net/qq_37761077/article/details/88750711
3.https://blog.csdn.net/linxigjs/article/details/79864033
4.https://pytorch.org/cppdocs/installing.html