CUDA error: CUBLAS_STATUS_NOT_INITIALIZED when calling cublasCreate(handle)

[JarintotionDin]

IndexKernel.cu:135: operator(): block: [13,0,0], thread: [16,0,0] Assertion index >= -sizes[i] && index < sizes[i] && "index out of bounds" failed.

File "/home/jarin/contrib/centermask2/centermask/modeling/centermask/center_heads.py", line 410, in forward
pred_instances = self.forward_with_given_boxes(features, proposals)
File "/home/jarin/contrib/centermask2/centermask/modeling/centermask/center_heads.py", line 438, in forward_with_given_boxes
instances = self._forward_maskiou(mask_features, instances)
File "/home/jarin/contrib/centermask2/centermask/modeling/centermask/center_heads.py", line 515, in _forward_maskiou
pred_maskiou = self.maskiou_head(mask_features, selected_mask)
File "/home/jarin/miniconda3/envs/pytorch1.8/lib/python3.7/site-packages/torch/nn/modules/module.py", line 744, in _call_impl
result = self.forward(*input, **kwargs)
File "/home/jarin/contrib/centermask2/centermask/modeling/centermask/maskiou_head.py", line 112, in forward
x = F.relu(self.maskiou_fc1(x))
File "/home/jarin/miniconda3/envs/pytorch1.8/lib/python3.7/site-packages/torch/nn/modules/module.py", line 744, in _call_impl
result = self.forward(*input, **kwargs)
File "/home/jarin/contrib/centermask2/centermask/layers/wrappers.py", line 86, in forward
x = super().forward(x)
File "/home/jarin/miniconda3/envs/pytorch1.8/lib/python3.7/site-packages/torch/nn/modules/linear.py", line 94, in forward
return F.linear(input, self.weight, self.bias)
File "/home/jarin/miniconda3/envs/pytorch1.8/lib/python3.7/site-packages/torch/nn/functional.py", line 1665, in linear
ret = torch.addmm(bias, input, weight.t())
RuntimeError: CUDA error: CUBLAS_STATUS_NOT_INITIALIZED when calling cublasCreate(handle)

[cuge1995]

Have you solve it?