原生PyTorch编译过程报错no module named yaml/typing_extensions

问题描述

PyTorch编译依赖yaml库和typing_extensions库,需要手动安装。

解决方案

  1. 执行以下命令安装依赖。

    pip3 install pyyaml
    pip3 install typing_extensions

  2. 安装依赖成功后,需要先执行make clean再执行bash build.sh进行编译,否则可能因缓存出现未知编译错误。