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

本文档详述了如何在Windows 10环境下,使用Anaconda和Visual Studio 2017从源码编译32位CPU版libtorch。由于官方不提供32位版本,因此需要自行编译。编译过程包括安装Anaconda、Visual Studio 2017,获取pytorch源码,设置环境变量,并测试编译后的libtorch。
摘要由CSDN通过智能技术生成

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版本后重新编译。


                
评论 72
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值