源码安装PyTorch

编译安装前准备

安装PyTorch依赖前请确保依赖列表中的OS依赖已全部安装。

安装系统依赖（以CentOS与Ubuntu操作系统为例）。

查看gcc和cmake版本。

gcc --version
cmake --version

要求7.3.0版本及以上gcc，如果版本低于7.3.0，请参见安装7.3.0版本gcc安装。若用户要安装1.11.0版本PyTorch，则gcc需为7.5.0版本以上。

要求3.12.0版本及以上cmake，如果版本低于3.12.0，请参见安装3.12.0版本cmake安装。

安装PyTorch 1.8.1或1.11.0

以下操作步骤以安装PyTorch 1.8.1版本为例。

1. 安装官方torch包。

   # 1.11.0版本请替换版本号为1.11.0+cpu
   pip3 install torch==1.8.1+cpu -f https://download.pytorch.org/whl/torch_stable.html --trusted-host download.pytorch.org

2. 编译生成PyTorch插件的二进制安装包。

   # 下载对应PyTorch版本分支代码，进入插件根目录，以v1.8.1-5.0.rc1为例，其他版本请替换对应版本号
   git clone -b v1.8.1-5.0.rc1 https://gitee.com/ascend/pytorch.git 
   cd pytorch    
   # 指定Python版本编包方式，以Python3.7为例，其他Python版本请使用 --python=3.8或--python3.9
   bash ci/build.sh --python=3.7

3. 安装pytorch/dist目录下生成的插件torch_npu包。

   # 请用户根据实际情况更改命令中的torch_npu包名
   pip3 install --upgrade dist/torch_npu-1.8.1.post1-cp38-cp38-linux_aarch64.whl

4. 安装对应框架版本的torchvision。

   #PyTorch 1.8.1需安装0.9.1版本，PyTorch 1.11.0需安装0.12.0版本
   pip3 install torchvision==0.9.1   

5. 配置环境变量，验证是否安装成功。
   1. 新建环境变量shell脚本env.sh，写入以下代码：

      # 配置CANN相关环境变量
      CANN_INSTALL_PATH_CONF='/etc/Ascend/ascend_cann_install.info'
      
      if [ -f $CANN_INSTALL_PATH_CONF ]; then
        DEFAULT_CANN_INSTALL_PATH=$(cat $CANN_INSTALL_PATH_CONF | grep Install_Path | cut -d "=" -f 2)
      else
        DEFAULT_CANN_INSTALL_PATH="/usr/local/Ascend/"
      fi
      
      CANN_INSTALL_PATH=${1:-${DEFAULT_CANN_INSTALL_PATH}}
      
      if [ -d ${CANN_INSTALL_PATH}/ascend-toolkit/latest ];then
        source ${CANN_INSTALL_PATH}/ascend-toolkit/set_env.sh
      else
        source ${CANN_INSTALL_PATH}/nnae/set_env.sh
      fi
      
      # 导入依赖库
      export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/usr/local/openblas/lib
      export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/usr/local/lib/
      export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/usr/lib64/
      export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/usr/lib/
      export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/usr/lib/aarch64_64-linux-gnu

      执行脚本，配置环境变量。

      source env.sh

   2. 执行如下命令，验证PyTorch是否安装成功。

      python -c "import torch;import torch_npu; a = torch.randn(3, 4).npu(); print(a + a);"

      显示如下回显证明PyTorch框架与插件安装成功。