youngwanLEE / centermask2

[CVPR 2020] CenterMask : Real-time Anchor-Free Instance Segmentation
Other
778 stars 159 forks source link

CUDA error: CUBLAS_STATUS_NOT_INITIALIZED when calling `cublasCreate(handle)` #74

Open JarintotionDin opened 3 years ago

JarintotionDin commented 3 years ago

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 commented 1 year ago

Have you solve it?