python - RuntimeError: CUDA 错误: CUBLAS_STATUS_EXECUTION_FAILED 当调用`cublasSgemm(handle, opa, opb, m, n, k, &alpha, a, lda, b, ldb, &beta, c, ldc)`时

Question

我正在尝试从 BertTopic 模型中提取羽毛，并且正在使用 jupyter notebook。第一个功能是情感分析，它通过以下代码完成： df-DataFrame, df["title_selftext"] - 此数据框列包含文本 (str)

  def get_sentiment(model, tokenizer, data):
     inputs = tokenizer(data, padding=True, return_tensors="pt")
     outputs = model(**inputs.to(device))
     return torch.nn.functional.softmax(outputs.logits.detach())

 res = []
 for txt in tqdm(batches(df["title_selftext"],20), total=len(df)//32):
    **res.append(get_sentiment(model,tokenizer, list(txt)))** 

当我试图为这个错误附加 get_sentiment 输出时： RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED 调用时cublasSgemm( handle, opa, opb, m, n, k, &alpha, a, lda, b, ldb, &beta, c, ldc)

请帮我解决这个错误谢谢