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