BlinkDL / RWKV-LM

RWKV is an RNN with transformer-level LLM performance. It can be directly trained like a GPT (parallelizable). So it's combining the best of RNN and transformer - great performance, fast inference, saves VRAM, fast training, "infinite" ctx_len, and free sentence embedding.
Apache License 2.0
12.05k stars 827 forks source link

windows平台下vscode运行RWKV-V4 train.py 报错 RuntimeError: Ninja is required to load C++ extensions #158

Closed psile closed 11 months ago

psile commented 1 year ago

已经下载Ninja,但是还是报错,有解决办法吗,小白求救

BlinkDL commented 11 months ago

Install VS2022 build tools (https://aka.ms/vs/17/release/vs_BuildTools.exe select Desktop C++). Reinstall CUDA 11.7 (install VC++ extensions). Run train.py in "x64 native tools command prompt".

Or you can install WSL2 Ubuntu