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
一开始是在加载数据之后添加的不管用,又在报错的行之前加了,就完事了