欢迎光临散文网 会员登陆 & 注册

win11 源码编译 pytorch CPU版

2023-04-29 14:48 作者:魔术iT  | 我要投稿

我本机安装这 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


win11 源码编译 pytorch CPU版的评论 (共 条)

分享到微博请遵守国家法律