记录调试中出现的问题:RuntimeError: CUDA error: CUBLAS_STATUS_INTERNAL_ERROR when calling cublasCreate(handle)
记录调试代码中出现的问题:RuntimeError: CUDA error: CUBLAS_STATUS_INTERNAL_ERROR when calling cublasCreate(handle)`论文链接:ANKGE(2023aaai)因为直接pip install -r requirements.txt, 会出现报错torch1.8.0+cu111无法找到,所以偷懒直接下载了torch1.8.0,结果导致运行到AnKnn.py的117行:trans_emb = torch.mm(rel_emb *
原创
2023-02-05 00:00:57 ·
840 阅读 ·
1 评论