PyTorch运行遇到找不到te问题

问题描述

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

解决措施

在开发环境下：

cd /usr/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 /usr/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