本地编译安装 Minkowski Engine 报错 Cuda 版本 与 Pytorch 版本不匹配

本文主要是介绍本地编译安装 Minkowski Engine 报错 Cuda 版本 与 Pytorch 版本不匹配，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

报错详情

RuntimeError:
The detected CUDA version (12.2) mismatches the version that was used to compile
PyTorch (11.8). Please make sure to use the same CUDA versions.

环境

• Ubuntu 22.04
• Python 3.8
• Cuda 12.2
• Pytorch cu118

报错分析

代码的 Readme 写明了 Cuda 版本 与 Pytorch 版本必须一致，但是其实 Pytorch cu118 是兼容 Cuda 12.2 的。
根据报错信息：

File "/home/littlespirit/anaconda3/envs/torch/lib/python3.8/site-packages/torch/utils/cpp_extension.py", line 525, in build_extensions_check_cuda_version(compiler_name, compiler_version)File "/home/littlespirit/anaconda3/envs/torch/lib/python3.8/site-packages/torch/utils/cpp_extension.py", line 413, in _check_cuda_versionraise RuntimeError(CUDA_MISMATCH_MESSAGE.format(cuda_str_version, torch.version.cuda))

可以看到，是编译过程中有函数在检查版本是否一致。
定位到 cpp_extension.py：525：

然后将 _check_cuda_version 函数注释掉，然后可以顺利使用兼容的 Pytorch 进入编译的下一阶段。

这篇关于本地编译安装 Minkowski Engine 报错 Cuda 版本 与 Pytorch 版本不匹配的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---