Pytorch与深度学习 —— 附录II. 如何在 Windows 中使用 MSVS 编写 Pytorch 程序

缺乏、安全感 2022-09-11 14:17 147阅读 0赞

> 如果你想用宇宙第一好用的 C/CXX 编译器 Visual Studio 来安装部署 PyTorch 的 Windows 应用，但是不知道该怎么办好，那么这一篇也许能告诉你答案。

### 文章目录 ###

*  前期准备
 *   *  Visual Studio
     *  LibTorch
     *  创建MSVC工程
     *  把LibTorch拷贝到工程目录下
 *  工程链接配置参数
 *  准备测试用例
 *  拷贝dll动态库

# 前期准备 #

## Visual Studio ##

尽管在 Windows 平台上也是可以通过部署CMake工程，来编译 Windows 版的 PyTorch 应用。但是我们放着宇宙第一好用的 Visual Studio 而使用其他的比如 CLion，Code::Blocks 这些工具，未免有些浪费了。毕竟现在Visual Studio都推出免费的社区版了，Debug又好用，安装配置项目也不麻烦，如果在Windows平台都还不使用Visual Studio写C代码，就像不用Eclipse写Java代码，不用PyCharm写Python代码一样，无聊的编程人的人生又要少几分色彩了。

如果在Windows平台还没装过Visual Studio的朋友，请一定通过以下链接去下载一个社区版的来玩一玩，你会感叹有钱的公司做的IDE就是不一样。

> https://visualstudio.microsoft.com/

![在这里插入图片描述][watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBA5omT56CB55qE6Zi_6YCa_size_20_color_FFFFFF_t_70_g_se_x_16_pic_center]  
如果你是个人开发者，选择Community就行，等待下载完成后，一路Next，大概半小时后就装好了。

## LibTorch ##

接下来我们需要LibTorch包，也就是PyTorch支持C/C++/Java开发的底层包，在官网可以直接下载。

> https://pytorch.org/

在当前1.9.0版本的时代，根据你的电脑具体配置情况下载对应的库包。如果你的PC支持CUDA运算，那么可以下载支持CUDA版本的库包。

![在这里插入图片描述][watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBA5omT56CB55qE6Zi_6YCa_size_20_color_FFFFFF_t_70_g_se_x_16_pic_center 1]

不过为了普适性，比如笔记本只有核显，或者是AMD家显卡的朋友，那么就直接下载CPU的版本就好。

## 创建MSVC工程 ##

安装好MSVS后，接下来需要创建一个MSVC工程，在Visual Studio的开始界面里，点击创建一个新工程。

![在这里插入图片描述][watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBA5omT56CB55qE6Zi_6YCa_size_16_color_FFFFFF_t_70_g_se_x_16_pic_center]  
然后在工程配置基础信息里，选择 Empty Project，并且决定好你的工程名字和工程代码存放路径。

![在这里插入图片描述][watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBA5omT56CB55qE6Zi_6YCa_size_20_color_FFFFFF_t_70_g_se_x_16_pic_center 2]  
![在这里插入图片描述][watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBA5omT56CB55qE6Zi_6YCa_size_20_color_FFFFFF_t_70_g_se_x_16_pic_center 3]  
然后点击Create按钮后，就创建好了。

## 把LibTorch拷贝到工程目录下 ##

工程创建完毕后，为了方便使用，最好把依赖库都拷贝到工程目录下。

![在这里插入图片描述][watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBA5omT56CB55qE6Zi_6YCa_size_20_color_FFFFFF_t_70_g_se_x_16_pic_center 4]

这样我们在配置工程目录链接时，就可以使用VS的一些默认参数，比如

> $(SolutionDir)

而可以不必使用完整路径，并且如果你习惯把工程代码拷贝到U盘，随身携带，这样在新的电脑里打开工程的时候，也不必重新配置一遍全部参数。

接下来，介绍下配置工程的外部链接所需要的具体参数。

# 工程链接配置参数 #

在工程的工作界面里，你需要右键点击一下项目名称

![在这里插入图片描述][watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBA5omT56CB55qE6Zi_6YCa_size_20_color_FFFFFF_t_70_g_se_x_16_pic_center 5]  
或者通过 Project/Properties  
![在这里插入图片描述][watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBA5omT56CB55qE6Zi_6YCa_size_20_color_FFFFFF_t_70_g_se_x_16_pic_center 6]  
打开工程属性配置界面

![在这里插入图片描述][watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBA5omT56CB55qE6Zi_6YCa_size_20_color_FFFFFF_t_70_g_se_x_16_pic_center 7]  
在 “VC++ Directories” 这个选项里，找到右侧的 “Include Directories” 和 “Library Directoies”。在这两项里，点击右侧一个下拉菜单，找到 “Edit” 选项。

![在这里插入图片描述][watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBA5omT56CB55qE6Zi_6YCa_size_20_color_FFFFFF_t_70_g_se_x_16_pic_center 8]  
请分别在 “Include Directories” 里，添加如下内容：

> $(SolutionDir)libtorch\\include  
> $(SolutionDir)libtorch\\include\\torch\\csrc\\api\\include

在 “Library Directories” 里添加如下内容

> $(SolutionDir)libtorch\\lib

编辑完成后，点击OK保存即可。

**需要注意一点的是，在配置工程平台信息（Platform）时，请选择x64平台，配置文件信息（Configuration）可以选择全平台通用（All Configurations）**

然后就是把工程生成文件类型，由默认的x86改为x64

![在这里插入图片描述][watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBA5omT56CB55qE6Zi_6YCa_size_20_color_FFFFFF_t_70_g_se_x_16_pic_center 9]  
这是因为LibTorch是x64版本的。

# 准备测试用例 #

接下来我们需要创建一个新的C++文件，并且写入下面的这段测试代码

#include <torch/torch.h>
    #include <iostream>
    
    #pragma comment(lib, "asmjit.lib")
    #pragma comment(lib, "c10.lib")
    #pragma comment(lib, "fbgemm.lib")
    #pragma comment(lib, "torch_cpu.lib")
    
    int main() { 
    	torch::Tensor tensor = torch::eye(3);
    	std::cout << tensor << std::endl;
    	return 0;
    }

点击编译按钮后，就会生成可执行文件。

# 拷贝dll动态库 #

然后我们需要做的就是把 libTorch里全部的dll文件都拷贝到生成的可执行文件目录下。当然你也可以只拷贝这几个：

> asmjit.dll  
> c10.dll  
> fbgemm.dll  
> libiomp5md.dll  
> torch\_cpu.dll

![在这里插入图片描述][watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBA5omT56CB55qE6Zi_6YCa_size_20_color_FFFFFF_t_70_g_se_x_16_pic_center 10]

最后再执行程序，就可以看到输出了。

![在这里插入图片描述][watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBA5omT56CB55qE6Zi_6YCa_size_20_color_FFFFFF_t_70_g_se_x_16_pic_center 11]

--------------------

如果你觉得这篇文章对你有用，点个赞再走呗！

[watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBA5omT56CB55qE6Zi_6YCa_size_20_color_FFFFFF_t_70_g_se_x_16_pic_center]: /images/20220828/2eb52e2890f34661a919bbb2de7fdd88.png
[watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBA5omT56CB55qE6Zi_6YCa_size_20_color_FFFFFF_t_70_g_se_x_16_pic_center 1]: /images/20220828/9d0e55b74d2144acbd5ee62c51fab291.png
[watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBA5omT56CB55qE6Zi_6YCa_size_16_color_FFFFFF_t_70_g_se_x_16_pic_center]: /images/20220828/5f632979abdb4728bea9bf8e4abd4245.png
[watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBA5omT56CB55qE6Zi_6YCa_size_20_color_FFFFFF_t_70_g_se_x_16_pic_center 2]: /images/20220828/e206d55d9a864013a939b201e4fa6d68.png
[watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBA5omT56CB55qE6Zi_6YCa_size_20_color_FFFFFF_t_70_g_se_x_16_pic_center 3]: /images/20220828/166152fa8fe9426297554b371e30dbc3.png
[watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBA5omT56CB55qE6Zi_6YCa_size_20_color_FFFFFF_t_70_g_se_x_16_pic_center 4]: /images/20220828/3781619bf4d04169b83c046816f3a17a.png
[watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBA5omT56CB55qE6Zi_6YCa_size_20_color_FFFFFF_t_70_g_se_x_16_pic_center 5]: /images/20220828/600e19c6a3634d6289d50589e02c918a.png
[watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBA5omT56CB55qE6Zi_6YCa_size_20_color_FFFFFF_t_70_g_se_x_16_pic_center 6]: /images/20220828/f6c4b035e8ae488ebbb7620195c08aac.png
[watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBA5omT56CB55qE6Zi_6YCa_size_20_color_FFFFFF_t_70_g_se_x_16_pic_center 7]: /images/20220828/a6354ac4076d42af9c962508f5fbe43c.png
[watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBA5omT56CB55qE6Zi_6YCa_size_20_color_FFFFFF_t_70_g_se_x_16_pic_center 8]: /images/20220828/2a85df2c201f49ecac24a2301cfaa047.png
[watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBA5omT56CB55qE6Zi_6YCa_size_20_color_FFFFFF_t_70_g_se_x_16_pic_center 9]: /images/20220828/91b95833e7ad495b9d25e1a321c0e474.png
[watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBA5omT56CB55qE6Zi_6YCa_size_20_color_FFFFFF_t_70_g_se_x_16_pic_center 10]: /images/20220828/74b7fa1018d240fd9b82698d9ff68d3c.png
[watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBA5omT56CB55qE6Zi_6YCa_size_20_color_FFFFFF_t_70_g_se_x_16_pic_center 11]: /images/20220828/26ebacb1096f42a3a3998fae4b4d5139.png