windwos源码编译libtorch(win10+32bit+libtorch)

忘是亡心i 2021-09-24 03:10 1288阅读 0赞

# win10 编译32位libtorch #

### 文章目录 ###

*  win10 编译32位libtorch
 *  前言
 *  1. 安装Anaconda
 *  2. 安装Visual Studio 2017
 *  3. 获取pytorch源码
 *  4. 设置环境变量
 *   *  1. 打开anaconda prompt
     *  2. 添加环境变量
 *  3. 编译好后libtorch的位置
 *  4. 测试libtorch
 *  5. 下载链接

# 前言 #

本人暂时没有试过GPU版的32位libtorch，所以以下编译出来的libtorch是基于cpu版的。同时由于目前pytorch官方并没有提供32位libtorch，如果项目需要32位libtorch必须使用源码进行编译。32位的libtorch在windows平台下因为不能使用mkl-dnn性能会有很大的降低，可以使用FBGEMM代替，但是我试了以后没有什么效果。要是有什么好的方式可以提升window平台下的速度可以告诉我，谢谢。

# 1. 安装Anaconda #

[anaconda下载地址][anaconda]  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MDQ0ODE0MA_size_16_color_FFFFFF_t_70]  
根据操作系统版本选择64位或32位的安装程序安装anaconda  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MDQ0ODE0MA_size_16_color_FFFFFF_t_70 1]  
安装时可以选择是否将anaconda添加到环境变量中，由于接下来我们要使用的是anaconda prompt来进行编译，所以这里可以根据需要随意选择。

# 2. 安装Visual Studio 2017 #

windows下源码编译pytorch需要用到vs，目前官方仅支持vs2017和vs2019，这里我使用vs2017来进行编译。  
[vs2017下载地址][vs2017]  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MDQ0ODE0MA_size_16_color_FFFFFF_t_70 2]  
勾选使用c++的桌面开发  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MDQ0ODE0MA_size_16_color_FFFFFF_t_70 3]  
选择单个组件，并勾选VC++2017版本15.4 v14.11工具集  
之后点击安装，等待安装完成。安装完成后将vs添加到环境变量中。  
参考: [Windows10下配置VS2017环境变量][Windows10_VS2017]  
如果配置环境变量时有些文件夹没有可以忽略。

# 3. 获取pytorch源码 #

`git clone -b`可以选择分支来克隆，这里我使用的是1.4.0的版本。如果在编译的时候出现问题，可以尝试更换pytorch版本后重新编译。

git clone -b v1.4.0 --recursive https://github.com/pytorch/pytorch
    cd pytorch
    git submodule sync
    git submodule update --init --recursive

# 4. 设置环境变量 #

## 1. 打开anaconda prompt ##

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

## 2. 添加环境变量 ##

# 如果设置为一，则编译出来的就是debug版本的libtorch
    set DEBUG=1
    # 将conda转换为32位环境
    set CONDA_FORCE_32BIT=1
    # 创建conda环境
    conda create -n env_name python=3.7
    # 进入conda环境
    conda activate env_name
    # 下载依赖包
    conda install numpy ninja pyyaml mkl mkl-include setuptools cmake cffi typing
    # 设置是否使用显卡，这里我设置为0，代表不使用显卡
    set USE_CUDA=0
    # 不适用ninja，使用ninja在编译时会发生问题，这里关闭ninja
    set USE_NINJA=OFF
    # （可选）是否编译test文件
    set BUILD_TEST=False
    # （可选）使用FBGEMM代替mkl-dnn编译
    set USE_FBGEMM=ON
    set CAFFE2=OFF
    # 设置你生成的makefile的文件版本
    set CMAKE_GENERATOR=Visual Studio 15 2017
    # 设置使用的工具集
    set CMAKE_GENERATOR_TOOLSET_VERSION=14.11
    set DISTUTILS_USE_SDK=1
    # 将编译环境转换为32位环境
    for /f "usebackq tokens=*" %i in (`"%ProgramFiles(x86)%\Microsoft Visual Studio\Installer\vswhere.exe" -version [15^,16^) -products * -latest -property installationPath`) do call "%i\VC\Auxiliary\Build\vcvarsall.bat" x86 -vcvars_ver=%CMAKE_GENERATOR_TOOLSET_VERSION%
    # 编译libtorch
    cd tools
    python build_libtorch.py

# 3. 编译好后libtorch的位置 #

你的pytorch位置下的torch文件夹下的bin、lib、include、share文件夹  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MDQ0ODE0MA_size_16_color_FFFFFF_t_70 5]  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MDQ0ODE0MA_size_16_color_FFFFFF_t_70 6]  
这里我漏标了一个include文件夹

# 4. 测试libtorch #

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MDQ0ODE0MA_size_16_color_FFFFFF_t_70 7]  
右键项目属性  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MDQ0ODE0MA_size_16_color_FFFFFF_t_70 8]  
修改包含目录和库目录  
![在这里插入图片描述][20200406164519786.png]  
包含目录(包含目录一般包含函数声明的头文件）  
![在这里插入图片描述][2020040616461081.png]  
库目录（一般包含静态库的位置）  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MDQ0ODE0MA_size_16_color_FFFFFF_t_70 9]  
**测试代码**

#include<torch/script.h>
    
    int main() { 
    	torch::Tensor tensor = torch::rand({ 1, 3, 224, 224});
    	std::cout << tensor << std::endl;
    	return 0;	
    }

# 5. 下载链接 #

[libtorch32位][libtorch32]

[anaconda]: https://www.anaconda.com/distribution/
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MDQ0ODE0MA_size_16_color_FFFFFF_t_70]: /images/20210923/cdee1c2bd50d40aab7c2cca5b20b9b58.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MDQ0ODE0MA_size_16_color_FFFFFF_t_70 1]: /images/20210923/450a96904aa74b85bca6e7775a3740fe.png
[vs2017]: https://visualstudio.microsoft.com/zh-hans/vs/older-downloads/
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MDQ0ODE0MA_size_16_color_FFFFFF_t_70 2]: /images/20210923/e2b10985c37a4ac1a014e6c257d5fc8d.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MDQ0ODE0MA_size_16_color_FFFFFF_t_70 3]: /images/20210923/ad49c7eb180d4eff974de6416d62b7df.png
[Windows10_VS2017]: https://blog.csdn.net/kaige_zhao/article/details/80315697
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MDQ0ODE0MA_size_16_color_FFFFFF_t_70 4]: /images/20210923/e7bdcc615e164e4f9fe0835f40a9a170.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MDQ0ODE0MA_size_16_color_FFFFFF_t_70 5]: /images/20210923/b517247177aa42c6aba287fefc5878c7.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MDQ0ODE0MA_size_16_color_FFFFFF_t_70 6]: /images/20210923/537cd439600446c3801e5d98ccd0a254.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MDQ0ODE0MA_size_16_color_FFFFFF_t_70 7]: /images/20210923/baf6d2efa9ee434880f1e68664ac8996.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MDQ0ODE0MA_size_16_color_FFFFFF_t_70 8]: /images/20210923/bb012760bef54043bd603ff6e4b278d0.png
[20200406164519786.png]: /images/20210923/79cd8727f802490cacfaac8ff7152334.png
[2020040616461081.png]: /images/20210923/a9b5824c598647da8fc47c87d16a8f94.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MDQ0ODE0MA_size_16_color_FFFFFF_t_70 9]: /images/20210923/f8b978cb50d14288b1b748abda25c760.png
[libtorch32]: https://download.csdn.net/download/weixin_40448140/12741972