CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling `cublasSgemm

CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling `cublasSgemm

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

CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling `cublasSgemm

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

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

torch.backends.cudnn.enabled = False

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

你可能感兴趣的:(Error,笔记)