源码安装PyTorch
编译安装前准备
安装PyTorch依赖前请确保依赖列表中的OS依赖已全部安装。
安装系统依赖(以CentOS与Ubuntu操作系统为例)。
openEuler、Kylin V10系统可参考CentOS进行安装。Debian系统可参考Ubuntu进行安装。
- CentOS
yum install -y patch libjpeg-turbo-devel dos2unix openblas git
- Ubuntu
apt-get install -y patch build-essential libbz2-dev libreadline-dev wget curl llvm libncurses5-dev libncursesw5-dev xz-utils tk-dev liblzma-dev m4 dos2unix libopenblas-dev git
查看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版本为例。
- 安装官方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
- 编译生成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
- 安装pytorch/dist目录下生成的插件torch_npu包。
# 请用户根据实际情况更改命令中的torch_npu包名 pip3 install --upgrade dist/torch_npu-1.8.1.post1-cp38-cp38-linux_aarch64.whl
- 安装对应框架版本的torchvision。
#PyTorch 1.8.1需安装0.9.1版本,PyTorch 1.11.0需安装0.12.0版本 pip3 install torchvision==0.9.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
- 执行如下命令,验证PyTorch是否安装成功。
python -c "import torch;import torch_npu; a = torch.randn(3, 4).npu(); print(a + a);"
显示如下回显证明PyTorch框架与插件安装成功。
- 新建环境变量shell脚本env.sh,写入以下代码:
父主题: 参考信息