使用DeBERTa模型在GPU运行报错: RuntimeError: CUDA error: CUBLAS_STATUS_NOT_INITIALIZED when calling cublasCrea

使用DeBERTa模型在GPU运行报错:

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

但用bert_base_chinese模型却是可以正常运行的,是什么原因呀?

model.to(device) # 此处报错

显示报错问题:

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

这是什么原因呀?该如何改动呀?

你可能感兴趣的:(深度学习,人工智能,bert)