当前位置：网站首页>源码编译pytorch坑

源码编译pytorch坑

2022-07-29 05:07:00 【要求不用太高】

nimble是一个对pytorch进行修改的神经网络加速包，本篇文章记录下配置nimble环境遇到的一些问题，可供以后回忆参考。

国内访问github很慢，而且由于需要递归拉去子模块：

git clone --recursive https://github.com/snuspl/nimble

经常fatal或者拉去失败，目前觉得原因是由于网络问题，解决方法就是不停的在项目目录中执行如下命令：

git submodule update --init --recursive

然后等执行编译脚本的时候根据报错提示前往第三方库中拉取对应的仓库到文件夹中

在pytorch的github的issue页面搜索得知，需要在编译时添加USE_NINJA=OFF选项，并且如果之前已经编过一遍，最好将build目录删掉重新编译

torchvision等库最好也从源码安装，才不会报错，此时我进行了从头编译与安装才没问题

进入终端，查看numpy版本号

pip show numpy

更新numpy

pip install -U numpy

解决

版权声明
本文为[要求不用太高]所创，转载请带上原文链接，感谢
https://blog.csdn.net/qq_44702267/article/details/125926017