RuntimeError: CUDA error: unspecified launch failure

299792459
Fri, 15 Oct 2021 19:24:37 +0000

Hi there,

Can someone tell me what’s wrong? I am getting a RuntimeError: CUDA error: unspecified launch failure. I have attached the output of python -m torch.utils.collect_env.

This occurs when I try to run:
trainer = Trainer(model=model,args=training_args,data_collator=data_collator,train_dataset=lm_datasets_train,eval_dataset=lm_datasets_valid)

Any help would be greatly appreciated. Thank you.

Read full topic