win10 编译32位libtorch
文章目录
前言
本人暂时没有试过GPU版的32位libtorch,所以以下编译出来的libtorch是基于cpu版的。同时由于目前pytorch官方并没有提供32位libtorch,如果项目需要32位libtorch必须使用源码进行编译。32位的libtorch在windows平台下因为不能使用mkl-dnn性能会有很大的降低,可以使用FBGEMM代替,但是我试了以后没有什么效果。要是有什么好的方式可以提升window平台下的速度可以告诉我,谢谢。
1. 安装Anaconda
anaconda下载地址
根据操作系统版本选择64位或32位的安装程序安装anaconda
安装时可以选择是否将anaconda添加到环境变量中,由于接下来我们要使用的是anaconda prompt来进行编译,所以这里可以根据需要随意选择。
2. 安装Visual Studio 2017
windows下源码编译pytorch需要用到vs,目前官方仅支持vs2017和vs2019,这里我使用vs2017来进行编译。
vs2017下载地址
勾选使用c++的桌面开发
选择单个组件,并勾选VC++2017版本15.4 v14.11工具集
之后点击安装,等待安装完成。安装完成后将vs添加到环境变量中。
参考: Windows10下配置VS2017环境变量
如果配置环境变量时有些文件夹没有可以忽略。
3. 获取pytorch源码
git clone -b
可以选择分支来克隆,这里我使用的是1.4.0的版本。如果在编译的时候出现问题,可以尝试更换pytorch版本后重新编译。