subprocess.CalledProcessError: Command '['ninja', '-v']' returned non-zero exit status 1.
问题:cuda版本不匹配或gcc设置不匹配
解决方法1:在.bashrc设置或者更改CUDA-HOME路径
ls -l /usr/local | grep cuda # 查看已安装cuda版本
nvcc --version # 查看当前 cuda 版本
#设置软链接到对应cuda版本
sudo rm -rf cuda
sudo ln -s /usr/local/cuda-11.1 /usr/local/cuda
在home/.bashrc 中添加: export CUDA_HOME=/usr/local/cuda
命令行运行: source ~/.bashrc
解决方法2:在原先语句 python setup.py build_ext --inplace
前添加编译器定义
运行 CC=gcc CXX=gcc python setup.py build_ext --inplace