CUDA error: CUBLAS_STATUS_NOT_INITIALIZED when calling cublasCreate(handle)

wzk1015 / video-bgm-generation

[ACM MM 2021 Best Paper Award] Video Background Music Generation with Controllable Music Transformer

https://wzk1015.github.io/cmt/

MIT License

291 stars 34 forks source link

CUDA error: CUBLAS_STATUS_NOT_INITIALIZED when calling cublasCreate(handle) #27

Closed MoshangLiu640 closed 5 months ago

MoshangLiu640 commented 6 months ago

你好，我在这个代码的基础上，增加了两个音乐属性的控制。我重新制作了数据集，使得训练集的形状从（3000,9999,9）变成了（3000,9999,11）。在模型中，我也增加了对应的loss和emb_size的维度，但是训练的时候报了这个错误。排查了一下，cuda版本和gpu memory没问题，似乎是tokenizer的问题。我想问一下这个问题该怎么解决？谢谢！