BlinkDL/RWKV-LM

Training on Cuda version 11.2, 11.3

cuongnguyengit opened this issue · 1 comments

Hi,

Please support me how to train rwkv model (rwkv4neo) in cuda version 11.2 or 11.3. I can't install 11.7 or newer.
Thanks.

what's the error you get