PyTorch运行遇到找不到te问题

问题描述

PyTorch运行过程中，提示缺少te依赖。

解决方案

开发环境

cd /urs/local/Ascend/ascend-toolkit/latest/{arch}-linux/lib64  #{arch}为架构名称。

pip3 install --upgrade topi-0.4.0-py3-none-any.whl

pip3 install --upgrade te-0.4.0-py3-none-any.whl

运行环境

cd /urs/local/Ascend/nnae/latest/{arch}-linux/lib64  #{arch}为架构名称。

pip3 install --upgrade topi-0.4.0-py3-none-any.whl

pip3 install --upgrade te-0.4.0-py3-none-any.whl