Pointnet2.pytorch出现的问题：RuntimeError:Command ‘[‘ninja‘, ‘-v‘]‘ returned non-zero exit status 1.

区区代码子

于 2023-02-15 14:06:53 发布

阅读量2.1k

点赞数 2

文章标签： python

本文链接：https://blog.youkuaiyun.com/qq_43903526/article/details/129042165

版权

根本原因是自己的pytorch版本和源代码所使用的pytorch版本不一致

解决：将ball_query.cpp、group_points.cpp、interpolate.cpp、sampling.cpp中的AT_CHECK 和 THCState_getCurrentStream(state)换成相应的 TORCH_CHECK 和 at::cuda::getCurrentCUDAStream()。
再python setup.py install 就ok了

其他尝试：

1、降低pytorch版本，但是服务器仅支持cuda11.0+版本，所以不行，并且pytorch3d对pytorch版本有要求

2、vim ~/.bashrc最后写：CUDA_HOME=/usr/local/cuda-11.3，但不行，根本问题在于pytorch版本

3、网上的更改ninja -v 变成ninja --version，没有尝试，但一看就不行，这是两个完全不同的命令，改了就无法编译了