问题产生原因
这个问题的出现与pytorch的版本有关,可以把pytorch版本降到1.5.0以下。但是这个解决方法治标不治本,有时候我们不可避免的需要使用高版本的pytorch,这就需要对pytorch做一些更改。
解决方法
修改anaconda环境下的lib/python3.6/site-packages/torch/utils/cpp_extension.py
文件,将将['ninja','-v']
改成['ninja','--v']
或者['ninja','--version']
。
但是这时候build会出现报错g++问题
,报错显示缺少xxx.o
文件。这时需要将别人成功编译好的xxx.o
文件放在自己的目录下。如果找不到别人编译好的文件,就需要自己在低版本pytorch下编译成功后获得该文件。