记录调试中出现的问题:RuntimeError: CUDA error: CUBLAS_STATUS_INTERNAL_ERROR when calling cublasCreate(handle)

记录调试代码中出现的问题:RuntimeError: CUDA error: CUBLAS_STATUS_INTERNAL_ERROR when calling cublasCreate(handle)`

这是调试2023(AAAI)ANKGE所遇到的问题

论文链接:ANKGE(2023aaai)

因为直接pip install -r requirements.txt, 会出现报错torch1.8.0+cu111无法找到,所以偷懒直接下载了torch1.8.0,结果导致运行到AnKnn.py的117行:trans_emb = torch.mm(rel_emb * ana_rel, self.trans_matrix) #[bs_train, dim]时不断报错:RuntimeError: CUDA error: CUBLAS_STATUS_INTERNAL_ERROR when calling cublasCreate(handle)。

起初以为是维度没有对齐,debug后发现维度没有问题,但是始终卡在这一行报错,后来想到cuda层的问题可能是torch或者cuda版本问题,所以查torch1.8.0+cu111怎么下载,通过以下方式下载了torch1.8.0+cu101替换了torch==1.8.0:

采用离线安装的方式 : 下载链接
选取和自己的环境对应的版本就行。

总结一下,还是得严格按照作者的环境流程来!

你可能感兴趣的:(代码调试问题记录,pytorch,深度学习,知识图谱,自然语言处理,人工智能)