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

GitHub 加速计划 / ni / ninja
6
0
下载
a small build system with a focus on speed
最近提交(Master分支:3 个月前 )
c74beb43 I think this also switches from Intel to ARM, let's see how it goes. 13 天前
4b72b15a Fix unit test crash, if test fails 1 个月前
Logo

AtomGit 是由开放原子开源基金会联合 CSDN 等生态伙伴共同推出的新一代开源与人工智能协作平台。平台坚持“开放、中立、公益”的理念,把代码托管、模型共享、数据集托管、智能体开发体验和算力服务整合在一起,为开发者提供从开发、训练到部署的一站式体验。

更多推荐