如何在昇腾910B上3步安装oil-gas-ops-prospectCANN 9.0环境配置完整指南【免费下载链接】oil-gas-ops-prospect面向油气勘探oil gas exploration领域的昇腾自定义算子库。仓名即 oil-gas-ops油气算子 prospect勘探目标面向地震成像、全波形反演等勘探计算场景项目地址: https://gitcode.com/cann/oil-gas-ops-prospectoil-gas-ops-prospect 是面向油气勘探领域的昇腾自定义算子库为地震成像、全波形反演等勘探计算场景提供ComplexMul、FusedBiasSoftmax、real_rfft等高性能算子。本文面向新手手把手讲解如何在Ascend 910B上完成CANN 9.0环境配置并用 3 步完成 oil-gas-ops-prospect 的安装与验证。一、安装前准备CANN 9.0 环境要求1.1 硬件与软件版本配套开始之前先确认环境满足以下要求完整说明见 docs/zh/context/quick_install.md项目要求CANN 版本9.0.0 官方版本Toolkit 包 对应 SoC 的 Ops 包加速卡Ascend 910BAscend910B1Atlas A2 系列Python≥ 3.10Python 依赖torch、torch-npu、Triton-Ascend 3.2.1已验证配套⚠️ 注意Triton-Ascend 必须使用与 CANN 配套的版本triton-ascend3.2.1不能用 PyPI 上的社区triton包代替——社区版没有昇腾后端。1.2 安装 CANN Toolkit 与 Ops 包按 CPU 架构x86_64/aarch64和产品型号从 CANN 下载中心选择对应的.run安装包依次执行# 1. 安装 Toolkit 包 chmod x Ascend-cann-toolkit_9.0.0_linux-x86_64.run ./Ascend-cann-toolkit_9.0.0_linux-x86_64.run --install --force --install-path/usr/local/Ascend # 2. 安装 Ops 包需与 Toolkit 装到同一目录910B 对应 910b chmod x Ascend-cann-910b-ops_9.0.0_linux-x86_64.run ./Ascend-cann-910b-ops_9.0.0_linux-x86_64.run --install --force --install-path/usr/local/Ascend1.3 配置环境变量并安装 Python 依赖# 加载 CANN 环境变量路径按实际安装位置替换 source /usr/local/Ascend/cann-9.0.0/set_env.sh # 安装项目 Python 依赖torch-npu 需与 CANN 严格配套 pip install -r requirements.txt # 安装 Triton-Ascend 3.2.1 pip uninstall -y triton triton-ascend pip install triton-ascend3.2.1 --extra-index-urlhttps://mirrors.huaweicloud.com/ascend/repos/pypi # 确认版本 python3 -c import importlib.metadata as m; print(m.version(triton-ascend))二、第 1 步下载 oil-gas-ops-prospect 源码git clone https://gitcode.com/cann/oil-gas-ops-prospect.git cd oil-gas-ops-prospect仓库按交付物形态分为三类算子后端安装一次全部可用后端目录交付物典型算子AscendCascendc/CANN OPP.run算子包aclnn接口ComplexMul、FusedBiasSoftmaxTriton-Ascendtriton-ascend/Python whl 内 Triton kernelreal_rfft/real_irfftComposite-Torchcomposite-torch/纯 PyTorch 组合模块Conv3d等价替换三、第 2 步一键编译并安装仓库根目录提供统一构建入口 build.sh推荐新手直接执行bash build.sh --install一条命令依次完成 5 件事编译 AscendC OPP.run算子包默认Ascend910B1默认全部算子将 OPP 包安装到当前 CANN 环境构建统一的oil_gas_ops_prospectPython whl 包通过pip安装 whl运行 whl 冒烟测试产物统一落在仓库根目录output/output/ ├── SOC/oil_gas_ops_prospect_opp_os_arch.run # OPP 算子包 └── wheels/oil_gas_ops_prospect-*.whl # Python whl安装完成后每个新 shell 都需要保留自定义 opapi 动态库路径建议写入~/.bashrcexport LD_LIBRARY_PATH${ASCEND_HOME_PATH}/opp/vendors/customize/op_api/lib:${ASCEND_HOME_PATH}/lib64:$LD_LIBRARY_PATH常用构建参数详见 docs/zh/context/build.mdbash build.sh --pkg # 只出 OPP .run 包 bash build.sh --opscomplex_mul # 只构建指定算子 bash build.sh --socAscend910B1 -j16 # 指定 SoC 与并行度 bash build.sh --clean # 先清理再构建 若 CANN 不是默认安装路径用环境变量指定ASCEND_HOME_PATH/path/to/cann bash build.sh --install四、第 3 步调用算子验证安装4.1 跑一个 Triton-Ascend 实数 FFT 算子import torch import torch_npu # noqa: F401 from oil_gas_ops_prospect.real_fft import real_irfft, real_rfft x torch.randn(8, 256, devicenpu) xr, xi real_rfft(x, dim-1) # 实数正向 FFT频率轴长度 129 y real_irfft(xr, xi, n256, dim-1) # 从半谱重建实信号4.2 跑 AscendC 复数逐点乘算子import os os.environ[OIL_GAS_OPS_COMPLEX_MUL_IMPL] ascendc # 默认走 torch 参考实现 import torch import torch_npu # noqa: F401 from oil_gas_ops_prospect.complex_mul import complex_mul ur, ui, vr, vi (torch.randn(4, 1024, devicenpu) for _ in range(4)) wr, wi complex_mul(ur, ui, vr, vi)更多可运行示例见 examples/ 目录接口签名与环境变量说明见 docs/zh/api_list.md。4.3 运行完整测试套件可选算子在 NPU 上的精度、性能必须在真实 910B 上验证bash build.sh -u # 全部测试套件 bash tests/run_whl_tests.sh # 只验证已安装的 whl测试分类说明见 tests/README.md。五、常见问题快速排查 ️现象原因与解决triton.backends.ascend导入失败误装了 PyPI 社区triton按 1.3 节两条命令重装triton-ascend3.2.1构建时提示 SoC 不受支持本仓当前仅验证Ascend910B1支持列表定义在 ascendc/scripts/config.py运行时报找不到 opapi 动态库未设置LD_LIBRARY_PATH见第 2 步末尾的命令torch-npu与 CANN 不匹配torch-npu必须与本地 CANN 版本严格配套勿从 PyPI 随意拉取环境调试精度与性能问题定位见 docs/zh/debug/op_debug_prof.md六、下一步 docs/QUICKSTART.md从环境准备到跑通算子的最短路径docs/zh/op_list.md算子数学定义、输入输出与约束docs/zh/invocation/quick_op_invocation.md更多算子调用方式docs/zh/develop/operator_development_guide.md新增自己的昇腾算子至此你已完成CANN 9.0 环境配置与oil-gas-ops-prospect 3 步安装下载源码 →build.sh --install→ 调用验证。后续新增算子或排查精度问题可随时回到本文第五节的排查表参考。【免费下载链接】oil-gas-ops-prospect面向油气勘探oil gas exploration领域的昇腾自定义算子库。仓名即 oil-gas-ops油气算子 prospect勘探目标面向地震成像、全波形反演等勘探计算场景项目地址: https://gitcode.com/cann/oil-gas-ops-prospect创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考