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

arch_list[-1] += '+PTX' IndexError: list index out of range #182

Closed theangle134 closed 9 months ago

theangle134 commented 10 months ago

when I run RWKV-v2/train,py or run.py, have a issues which is 'arch_list[-1] += '+PTX' IndexError: list index out of range'. please how can I solve this problem?

BlinkDL commented 9 months ago

seems your CUDA is not installed correctly