编译libtorch_npu
- 参见安装前必读、安装PyTorch框架和安装torch_npu插件完成依赖与PyTorch框架的安装。
- 获取libtorch_npu源码。v2.0.1-5.0.0指支持PyTorch 2.0.1版本的v2.1.0-5.0.0分支,其他版本可参考表1修改命令。
git clone -b v2.0.1-5.0.0 https://gitee.com/ascend/pytorch.git cd pytorch git submodule update --init --recursive
- 执行编译生成libtorch_npu安装包。
python3 build_libtorch_npu.py
默认编译release版本,如需debug版本,添加DEBUG=1环境变量。编译完成后,当前目录下生成libtorch_npu目录,包含以下文件。
- include:生成的C++头文件。
- lib:生成的C++库文件。
- share:包含Torch_npuConfig.cmake,用于用户编译构建时获取必要的头文件,库文件等配置文件。
libtorch推理测试,请参见libtorch推理。
父主题: 安装PyTorch