本文主要是介绍本地编译安装 Minkowski Engine 报错 Cuda 版本 与 Pytorch 版本不匹配,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
编译 Cuda 版本 C++ 插件 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 版本不匹配的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!