Win11编译cpu版本的pytorch源码

2023-11-08

1、下载源码和依赖库

      首先,在github上下载指定版本的pytorch源码,地址如下:https://github.com/pytorch/pytorch,因为在pytorch源码中有一些依赖库是直接给出的超链接,因此都需要直接离线下载依赖库。pytorch依赖库均下载到third_parth目录中。

2、编译pytorch源码

     编译环境win11,anaconda3。

2.1 创建虚拟环境

首先,利用conda创建虚拟环境:

conda create -n pytorch1.10 python==3.7

进入虚拟环境中:

conda activate pytorch1.10

查看是否创建成功:

2.2  编译

首先,进入pytorch源码目录下,进入终端,之后输入cmd,因为是cpu版本直接设置USE_CUDA=0,设置REL_WITH_DEB_INFO=1,否则windows环境下无法进入C++ debug模式。

最后,执行python setup.py develop开始执行编译。

编译过程中的出现的error:

1.could not find any instance of Visual Studio

   报这个错是因为windows上没有适配到MSVC编译器,我们需要在path中添加这MSCV,具体操作可以参考这个博客,Visual Studio 2019 配置 MSVC 环境变量,使用命令行编译。 - 简书 (jianshu.com)

 之后,输入自己所在的dat文件:

"C:\Program Files\Microsoft Visual Studio\2022\Community\VC\Auxiliary\Build\vcvarsall.bat" x64 -vcvars_ver=14.3

 

 

本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

Win11编译cpu版本的pytorch源码 的相关文章

随机推荐