配置环境后,运行代码提示错误
global2keyego = torch.inverse(keyego2global.float()) # (B, 1, 4, 4)
RuntimeError: cusolver error: CUSOLVER_STATUS_INTERNAL_ERROR, when calling `cusolverDnCreate(handle)`
直接运行python代码,也会出错。
怀疑应该是pytorch版本与cuda版本不一致的问题
cuda-11.7
pytorch安装1.10.0+cu111
使用如下指令可解决问题
pip install torch==1.10.0+cu113 torchvision==0.11.0+cu113 torchaudio==0.10.0 -f https://download.pytorch.org/whl/torch_stable.html