安装前必读

前提条件

• 安装配套版本的CANN软件并配置CANN环境变量。
• 安装二进制算子包具体请参考安装二进制算子包。
• 安装对应框架版本的torchvision，PyTorch 1.11.0需安装0.12.0版本，PyTorch 2.0.1版本需安装0.15.2版本，PyTorch 2.1.0版本需安装0.16.0版本。
  执行如下命令安装torchvision，以0.12.0版本为例：

  pip3 install torchvision==0.12.0 

• 通过源码编译安装torch_npu插件时，安装如下环境依赖。
  执行如下命令安装。如果使用非root用户安装，需要在命令后加--user，例如：pip3 install pyyaml --user。

  pip3 install pyyaml
  pip3 install wheel
  pip3 install setuptools

版本配套关系

安装步骤

安全提示：建议使用非管理员账户安装运行torch_npu，且建议对安装程序的目录文件做好权限管控，建议文件夹权限设置为750，文件权限设置为 640。可以通过设置umask控制安装后文件的权限，如设置umask为0027。更多安全相关内容请参考安全说明。

在昇腾环境安装PyTorch步骤如下：

1. 安装PyTorch框架。
2. 安装torch_npu插件：可以直接下载whl包快速安装也可以通过源码编译安装。
3. （可选）编译libtorch_npu：需要进行libtorch推理测试时执行。
4. （可选）安装APEX模块：混合精度训练时安装。

• 如果用户不确定自己的软件包版本，可参考查询软件包版本信息进行查询。