原生PyTorch编译过程报错no module named yaml/typing_extensions
问题描述
PyTorch编译依赖yaml库和typing_extensions库,需要手动安装。
解决方案
执行以下命令安装依赖。
pip3 install pyyaml pip3 install typing_extensions
安装依赖成功后,需要先执行
make clean
再执行
bash build.sh
进行编译,否则可能因缓存出现未知编译错误。
父主题:
FAQ