cann | amct量化环境安装_amct onnx-CSDN博客

本文链接：https://blog.csdn.net/m0_59237910/article/details/136991097

本文详细介绍了如何通过conda创建和管理Python环境，包括安装amct环境，onnx和onnxruntime，以及如何编译和验证自定义算子包的过程。特别提到了不同版本的依赖匹配和注意事项。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

# (1) Anacona3安装，用conda管理python环境，略过

# (2) conda创建amct环境，python版本3.7.5 ==> 3.7.5通，3.9.2不通
conda create -n amct_py375 python=3.7.5

# (3) 激活conda环境
conda activate amct_py375

# (4) 安装onnx=1.8.0，onnxruntime=1.6.0，protobuf=3.11.3 ==> 注意版本号
pip install -i https://pypi.douban.com/simple onnx=1.8.0 onnxruntime=1.6.0 protobuf=3.11.3

# (5) 下载Ascend-cann-amct_{software version}_{arch}.tar.gz并解压
#     下载地址：https://www.hiascend.com/software/cann/community-history
tar -zxvf Ascend-cann-amct_{software version}_{arch}.tar.gz

# (6) 安装amct_onnx-{version}-py3-none-linux_{arch}.whl ==> 以onnx为例
pip install amct_onnx-{version}-py3-none-linux_{arch}.whl

# (7) 编译并安装自定义算子包
tar -zxvf amct_onnx_op.tar.gz
## 解压后目录
## amct_onnx_op
## |--inc
## |--src
## |--setup.py
cd amct_onnx_op
python setup.py build

# (8) 验证amct_onnx是否安装成功
amct_onnx
# 或
>> python
>> import amct_onnx

需要注意的是：在编译自定义算子包前，在amct_onnx_op/inc里面还需要手动下载四个头放进去，下载路径：https://github.com/microsoft/onnxruntime/tree/v1.6.0/include/onnxruntime/core/session (注意版本的对应，否则在import amct_onnx时会出现segmentation fault)。四个头包括：

onnxruntime_cxx_api.h
onnxruntime_cxx_inline.h
onnxruntime_c_api.h
onnxruntime_session_options_config_keys.h