我在A40显卡上运行的代码,一直报错!
环境:torch 1.8.1+cu111
torchstat 0.0.7
torchvision 0.9.1+cu111
python 3.8
File "/root/autodl-tmp/ICCV-2023-MB-TaylorFormer-main/basicsr/models/archs/MB_TaylorFormer.py", line 253, in forward
x = x + self.attn(self.norm1(x))
File "/root/miniconda3/lib/python3.8/site-packages/torch/nn/modules/module.py", line 889, in _call_impl
result = self.forward(*input, **kwargs)
File "/root/autodl-tmp/ICCV-2023-MB-TaylorFormer-main/basicsr/models/archs/MB_TaylorFormer.py", line 229, in forward
out_numerator = torch.sum(v, dim=-2).unsqueeze(2)+(q@attn)
RuntimeError: CUDA error: CUBLAS_STATUS_INTERNAL_ERROR when calling cublasSgemmStridedBatched( handle, opa, opb, m, n, k, &alpha, a, lda, stridea, b, ldb, strideb, &beta, c, ldc, stridec, num_batches)
麻烦您看一下,谢谢!
我在A40显卡上运行的代码,一直报错! 环境:torch 1.8.1+cu111 torchstat 0.0.7 torchvision 0.9.1+cu111 python 3.8 File "/root/autodl-tmp/ICCV-2023-MB-TaylorFormer-main/basicsr/models/archs/MB_TaylorFormer.py", line 253, in forward x = x + self.attn(self.norm1(x)) File "/root/miniconda3/lib/python3.8/site-packages/torch/nn/modules/module.py", line 889, in _call_impl result = self.forward(*input, **kwargs) File "/root/autodl-tmp/ICCV-2023-MB-TaylorFormer-main/basicsr/models/archs/MB_TaylorFormer.py", line 229, in forward out_numerator = torch.sum(v, dim=-2).unsqueeze(2)+(q@attn) RuntimeError: CUDA error: CUBLAS_STATUS_INTERNAL_ERROR when calling
cublasSgemmStridedBatched( handle, opa, opb, m, n, k, &alpha, a, lda, stridea, b, ldb, strideb, &beta, c, ldc, stridec, num_batches)
麻烦您看一下,谢谢!