CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling `cublasSgemm

野性酷女 2023-01-01 10:59 357阅读 0赞

CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling `cublasSgemm

在模型测试的时候出现了这个问题,

  1. CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling `cublasSgemm

由于训练的时候没问题,就把cuda版本、python版本什么的原因直接排除了

参考了这篇的第一个评论:

torch.backends.cudnn.enabled = False

一开始是在加载数据之后添加的不管用,又在报错的行之前加了,就完事了

发表评论

表情:
评论列表 (有 0 条评论,357人围观)

还没有评论,来说两句吧...

相关阅读