“Ninja is required to load C++ extensions”解决方案（Pychharm会 shell执行没这问题）

最新推荐文章于 2025-02-18 16:49:24 发布

DanielFaster

最新推荐文章于 2025-02-18 16:49:24 发布

阅读量441

点赞数 1

CC 4.0 BY-SA版权

分类专栏：深度学习 python 文章标签： c++ 深度学习人工智能

本文链接：https://blog.csdn.net/binbinczsohu/article/details/144447146

python 同时被 2 个专栏收录

28 篇文章

订阅专栏

深度学习

8 篇文章

订阅专栏

问题描述

from torch.utils.cpp_extension import load
wkv_cuda = load(name="wkv", sources=["./cuda/wkv_op.cpp", "./cuda/wkv_cuda.cu"],
                verbose=True, extra_cuda_cflags=['-res-usage', '--maxrregcount 60', '--use_fast_math', '-O3', '-Xptxas -O3', f'-DTmax={T_MAX}'])