安装指南

本教程面向使用 Accelerate & 昇腾的开发者，帮助完成昇腾环境下 Accelerate 的安装。

Accelerate 下载安装

备注

阅读本篇前，请确保已按照 安装教程 准备好昇腾环境！ 或者直接使用具备昇腾环境的镜像 cosdt/cann:8.0.rc1-910b-ubuntu22.04, 更多的版本可至 cosdt/cann 获取。

启动镜像

docker run -itd --network host -v /usr/local/dcmi:/usr/local/dcmi -v /usr/local/bin/npu-smi:/usr/local/bin/npu-smi -v /usr/local/Ascend/driver:/usr/local/Ascend/driver -v /etc/ascend_install.info:/etc/ascend_install.info --device /dev/davinci7 --device /dev/davinci_manager --device /dev/devmm_svm --device /dev/hisi_hdc --shm-size 16G --name accelerate cosdt/cann:8.0.rc1-910b-ubuntu22.04 bash

安装 Accelerate 及依赖包

pip install torch==2.2.0 torch_npu==2.2.0 accelerate -i https://pypi.tuna.tsinghua.edu.cn/simple