编译libtorch_npu

  1. 参见安装前必读安装PyTorch框架安装torch_npu插件完成依赖与PyTorch框架的安装。
  2. 获取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

  3. 执行编译生成libtorch_npu安装包。

    python3 build_libtorch_npu.py

    默认编译release版本,如需debug版本,添加DEBUG=1环境变量。编译完成后,当前目录下生成libtorch_npu目录,包含以下文件。

    • include:生成的C++头文件。
    • lib:生成的C++库文件。
    • share:包含Torch_npuConfig.cmake,用于用户编译构建时获取必要的头文件,库文件等配置文件。

    libtorch推理测试,请参见libtorch推理