[Tools: ninja] RuntimeError: Ninja is required to load C++ extension

文章提供了在Linux环境中安装Ninja的两种方法,针对无root权限的情况推荐使用pip安装,并说明了如何在PyCharm中配置Ninja路径。此外,还指出了当遇到nvccfatal:Unsupportedgpuarchitecturecompute_89错误时,可能是因为CUDA版本过低,升级CUDA到11.8可以解决此问题。
摘要由CSDN通过智能技术生成

1. 安装Ninja

  • 如果有root权限,可以下载ninja并安装:
wget https://github.com/ninja-build/ninja/releases/download/v1.8.2/ninja-linux.zip
sudo unzip ninja-linux.zip -d /usr/local/bin/
sudo update-alternatives --install /usr/bin/ninja ninja /usr/local/bin/ninja 1 --force 
  • 如果没有root权限,则通过pip安装:
pip install ninja

2. Linux可以正常运行,Pycharm运行则继续报错

  • 如果是通过pip安装,则将ninja包的路径显式添加:
import os
os.environ['PATH'] += ':/path/to/conda/envs/xxx/bin/'

报错:nvcc fatal : Unsupported gpu architecture 'compute_89'

1. cuda版本过低,从cuda-11.6升级至cuda-11.8可以解决

# 查看当前nvcc支持的gpu框架
which nvcc
nvcc --list-gpu-arch

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值