win11 源碼編譯 pytorch CPU版
我本機(jī)安裝這 visual studio 2022 所以官方要求那個(gè) msvc 工具 就沒裝了
項(xiàng)目是bazel 的 所以需要下載bazel 并將bazel 加入 PATH 中
bazel的版本需要看 pytorch中 .bazelversion 的版本 :
下載地址:
https://github.com/bazelbuild/bazel/tags
下載源碼:要遞歸下載 項(xiàng)目依賴很多庫
git clone --recursive https://github.com/pytorch/pytorch.git
如果斷開了 使用下面命令繼續(xù)下載 直到所有庫完成為止
git submodule sync
git submodule update --init --recursive
要安裝GPU 需要 安裝 cuda (我沒有用這個(gè) 所以 這個(gè)參考官方安裝)
下載完成后(使用這個(gè)安裝依賴工具):
pip install -r requirements.txt
pip install cmake ninja
還需要安裝些其他依賴工具
pip install numpy pyyaml mkl mkl-include setuptools cffi
編譯預(yù)設(shè)(CMD 中 執(zhí)行 set 就行了 )
這個(gè)CMD是vsual studio中的生成工具x64 Native Tools Command Prompt for VS 2022
使用 cmd 那個(gè)窗口是不行的:
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