Files
openmind/docs/zh/install.md
白超 ea3a0e073f !240 修改silicondiff_npu版本信息
Merge pull request !240 from 白超/master
2025-06-17 06:46:09 +00:00

7.7 KiB
Raw Permalink Blame History

安装

软件版本配套

openMind Library v1.0.0版本配套说明如下目前仅支持Linux系统。

产品名称 产品版本 获取链接
MindSpore 2.4.0 https://www.mindspore.cn/versions#2.4.0
PyTorch 2.1.0 https://download.pytorch.org/whl/torch/
PyTorch-NPU 6.0.rc3-pytorch2.1.0 https://gitee.com/ascend/pytorch/releases/tag/v6.0.rc3-pytorch2.1.0
CANN 8.0.RC3.beta1 https://www.hiascend.com/developer/download/community/result?module=cann&cann=8.0.RC3.beta1
HDK 1.0.26.alpha https://www.hiascend.com/hardware/firmware-drivers/community?product=6&model=27&cann=8.0.RC3.alpha003&driver=1.0.26.alpha
HDKRC3版本可选 1.0.27.alpha https://www.hiascend.com/hardware/firmware-drivers/community?product=4&model=26&cann=8.0.RC3.beta1&driver=1.0.27.alpha
MindSpeed可选 1.0.RC2/ https://gitee.com/ascend/MindSpeed/tree/1.0.RC2/
Megatron-LM可选 0.6.0 https://github.com/NVIDIA/Megatron-LM/releases/tag/core_v0.6.0
MindSpore NLP可选 0.4.1 https://github.com/mindspore-lab/mindnlp/tree/v0.4.1
silicondiff_npu可选 2.1.0.post3 https://pypi.org/project/silicondiff-npu/2.1.0.post3
mindone可选 0.2.0 https://gitee.com/mindspore-lab/mindone/tree/v0.2.0/

安装指导

由于仅支持昇腾NPU处理器因此以下操作均在NPU环境上执行。

步骤一:驱动和固件安装

  • 首次安装,先安装驱动再安装配套固件

    首次安装场景:环境初始未安装驱动,npu-smi查不到npu信息或者lsmod | grep drv查不到驱动模块。

  • 升级场景,先安装固件再安装配套驱动

    升级场景:环境初始已安装驱动,npu-smi可正常查询到设备信息。

以升级场景为例安装HDK。

bash Ascend-hdk-<soc_version>-npu-firmware_*.run --full
bash Ascend-hdk-<soc_version>-npu-driver_*.run --full

reboot

详细的安装教程可参考:安装驱动和固件

步骤二CANN包安装

根据设备类型是x86或aarch64下载对应的CANN包。CANN包默认安装到/usr/local/Ascend下也可以选择安装在指定的路径下需要加上--install-path参数注意整个路径权限都要755以上。

以安装到指定路径下为例安装CANN包。

#创建安装目录
mkdir -p /xxx/your_folder/cann/
chmod -R 755 /xxx/your_folder

#安装CANN包
bash Ascend-cann-toolkit_*_linux-*.run --install-path=/xxx/your_folder/cann/ --full
bash Ascend-cann-kernels-*_linux.run --install-path=/xxx/your_folder/cann/ --install

#source环境变量
source /xxx/your_folder/cann/ascend-toolkit/set_env.sh

详细的安装教程可参考:安装CANN

步骤三:环境准备

在进行openMind Library的安装之前请确保您已经安装了Python以及openMind Hub Client。当前已支持的版本及安装说明如下

步骤四:虚拟环境安装指导

为避免不同项目间不同依赖项的兼容性问题推荐在Python或conda虚拟环境中安装openMind Library。有关虚拟环境的安装和管理请详见Python虚拟环境conda虚拟环境此处仅提供创建和启用虚拟环境的方法以Python 3.10版本为例:

  • Python虚拟环境创建与使用方式。

    注意此种安装方式的前提是系统中已安装Python 3.10。

    # 创建名为your_venv_name的python虚拟环境
    python3 -m venv /path/to/your_venv_name
    # 激活your_venv_name虚拟环境
    source /path/to/your_venv_name/bin/activate
    
  • conda虚拟环境创建与使用方式。

    # 创建名为your_venv_name的python 3.10的虚拟环境
    conda create -n your_venv_name python=3.10
    # 激活your_venv_name虚拟环境
    conda activate your_venv_name
    

由于conda的跨平台特性更方便不同平台用户的使用因此更推荐用户选择conda来管理虚拟环境。

步骤五pip安装openMind Library

openMind Library可以通过pip进行安装,请根据实际环境选择对应的命令进行安装。

需要注意的是,由于torch-npu依赖的torchaarch64下可通过pip直接安装,但是在x86下需要设置特定url下载cpu版本,所以两个环境下安装命令有所不同,具体安装代码在下文已做区分呈现。

  • 不安装框架仅安装openMind Library可体验魔乐社区上传和下载功能。

    pip install openmind
    
  • 安装openMind Library并安装PyTorch框架及其依赖。

    #aarch64环境下安装
    pip install openmind[pt]
    
    #x86环境下安装
    pip install openmind[pt] --extra-index-url https://download.pytorch.org/whl/cpu        
    
  • 安装openMind Library并安装MindSpore框架及其依赖。

    pip install openmind[ms]
    

步骤六:安装结果检验

通过上述方法安装openMind Library后可执行以下命令检验openMind Library是否成功安装

python -c "import openmind;print(openmind.__version__);"

该命令成功执行并在终端打印出安装openMind Library的版本号则表示您成功安装了openMind Library。如下图为成功安装openMind Library后的回显示例具体版本信息随演进变化

openMind Library安装成功回显

卸载

使用pip完成openMind Library包的卸载

pip uninstall openmind

提示

通过以下指令获取安装/卸载openMind Library日志其中LOG_FILE为用户指定的日志路径或日志文件名。

【注意】安装/卸载日志不会转储,安装前请注意磁盘空间剩余容量。

# 获取安装openMind Library日志
pip install openmind --log LOG_FILE
# 获取卸载openMind Library日志
pip uninstall openmind --log LOG_FILE