RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`

HRNet / HRNet-Semantic-Segmentation

The OCR approach is rephrased as Segmentation Transformer: https://arxiv.org/abs/1909.11065. This is an official implementation of semantic segmentation for HRNet. https://arxiv.org/abs/1908.07919

Other

3.09k stars 682 forks source link

RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)` #259

Open Gkkkkkkkkk opened 2 years ago

Gkkkkkkkkk commented 2 years ago

When running on Cityscapes, this error happens in "seg_hrnet_ocr.py" line 65. Does anyone know how to solve this?