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

SkyeSong38 / CSWinTT

Transformer Tracking with Cyclic Shifting Window Attention (CSWinTT)

MIT License

68 stars 6 forks source link

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

Open Quan-w opened 2 years ago

Quan-w commented 2 years ago

During the training, I am getting this error, but if it is inference then there is no problem.

File "F:\transformer\CSWinTT\lib\train../..\lib\models\cswintt\transformer_cs.py", line 256, in forward _query, _key, _value = self.qkv_embedding(merge_feat).chunk(3, dim=-1)

Quan-w commented 2 years ago

跑stark和mixformer就没用出现这个错误