RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling `cublasSgemv

在进行项目复现时遇到如下错误：

img


如果不用GPU训练，那么在运行一段时间后，将报错已杀死。预测与维度无关。
环境如下：python==3.7，ubuntu==20.04

img