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

[Quan-w]

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]

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