win11 源码编译 pytorch CPU版
我本机安装这 visual studio 2022 所以官方要求那个 msvc 工具 就没装了
项目是bazel 的 所以需要下载bazel 并将bazel 加入 PATH 中
bazel的版本需要看 pytorch中 .bazelversion 的版本 :
下载地址:
https://github.com/bazelbuild/bazel/tags
下载源码:要递归下载 项目依赖很多库
git clone --recursive https://github.com/pytorch/pytorch.git
如果断开了 使用下面命令继续下载 直到所有库完成为止
git submodule sync
git submodule update --init --recursive
要安装GPU 需要 安装 cuda (我没有用这个 所以 这个参考官方安装)
下载完成后(使用这个安装依赖工具):
pip install -r requirements.txt
pip install cmake ninja
还需要安装些其他依赖工具
pip install numpy pyyaml mkl mkl-include setuptools cffi
编译预设(CMD 中 执行 set 就行了 )
这个CMD是vsual studio中的生成工具x64 Native Tools Command Prompt for VS 2022
使用 cmd 那个窗口是不行的:
set DEBUG=1 # 编译 debug 版
set USE_CUDA=0 # CPU 版本
set REL_WITH_DEB_INFO=1
开始编译:
python setup.py develop
只编译 libtorch:
cd tools
python build_libtorch.py