hc-skadi/ascendc/cann-env-setup

npu-smi info

# 检查Python
if ! command -v python3 &> /dev/null && ! command -v python &> /dev/null; then
    echo "Python not found, please install from env_dependence.md"
fi
 
# 检查pip
if ! command -v pip3 &> /dev/null && ! command -v pip &> /dev/null; then
    echo "pip not found, please install from env_dependence.md"
fi

# 检查当前文件夹是否存在CANN安装包
if ls Ascend-cann-toolkit_*.run 2>/dev/null | grep -q .; then
    echo "找到CANN Toolkit安装包"
    TOOLKIT_PACKAGE=$(ls Ascend-cann-toolkit_*.run)
else
    echo "未找到CANN Toolkit安装包，请使用在线安装方式"
    # 退出离线安装，建议使用在线安装
    exit 1
fi
 
# 检查当前文件夹是否存在ops安装包
if ls Ascend-cann-*-ops_*.run 2>/dev/null | grep -q .; then
    echo "找到CANN ops安装包"
    OPS_PACKAGE=$(ls Ascend-cann-*-ops_*.run)
else
    echo "未找到CANN ops安装包，请使用在线安装方式"
    # 退出离线安装，建议使用在线安装
    exit 1
fi

# 安装 Toolkit (默认路径 /usr/local/Ascend)
bash "$TOOLKIT_PACKAGE" --install --quiet
# 安装 ops
bash "$OPS_PACKAGE" --install --quiet

# 创建Python环境（CANN支持Python 3.7.x - 3.13.11）
conda create -n cann_env python
# 激活环境
conda activate cann_env

# 添加昇腾conda源
conda config --add channels https://repo.huaweicloud.com/ascend/repos/conda

# 使用conda安装CANN工具包
conda install ascend::cann-toolkit
# 安装CANN ops包（根据芯片类型选择，如910、910b、310等）
conda install ascend::cann-{芯片类型}-ops

sudo curl https://repo.oepkgs.net/ascend/cann/ascend.repo -o /etc/yum.repos.d/ascend.repo && yum makecache

yum install -y Ascend-cann-toolkit
yum install -y Ascend-cann-{芯片类型}-ops

source /usr/local/Ascend/ascend-toolkit/set_env.sh

echo "source /usr/local/Ascend/ascend-toolkit/set_env.sh" >> ~/.bashrc

python3 -c "import acl;acl.init();acl.rt.set_device(0)"

pip3 install attrs cython numpy decorator sympy cffi pyyaml pathlib2 psutil protobuf==3.20.0 scipy requests absl-py --user
pip3 install protobuf==3.20.0