源码构建
环境准备
在源码编译前,请先完成基础环境搭建,具体操作请参见快速安装。
源码编译
下载源码
开发者可通过如下命令下载本仓源码:
# 下载项目源码,以master分支为例
git clone https://gitcode.com/cann/amct.git
编译
进入项目根目录,执行如下编译命令,生成 amct_pytorch 安装包:
bash build.sh --torch
编译成功后,会在项目根目录的build_out目录下生成amct_pytorch-${version}-py3-none-linux_${arch}.tar.gz。
- ${version}表示版本号。
- ${arch}表示CPU架构,如aarch64、x86_64。
更多特性请参考:https://hiascend.com/document/redirect/CannCommunityToolAmct
本地验证
利用tests路径下的测试用例进行本地验证:
-
安装依赖
# 安装测试依赖 pip3 install pytest-cov -
执行测试用例:
bash build.sh -u更多执行选项可以用 -h 查看:
bash build.sh -h
安装与卸载
-
安装编译环节生成的安装包(如果安装用户为root,请将安装命令中的--user删除)。
pip3 install amct_pytorch-${version}-py3-none-linux_${arch}.tar.gz --user说明
安装AMCT工具时,请确保pip版本<=25.2,否则可能出现“ModuleNotFoundError:No module named 'torch' ”错误信息;如果用户pip版本>25.2,且不想降低版本,则请在上述安装命令后追加
--no-build-isolation。 -
卸载
pip3 uninstall amct_pytorch
安装完成后,可以参考样例运行运行样例。