本地编译安装 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_version
    raise RuntimeError(CUDA_MISMATCH_MESSAGE.format(cuda_str_version, torch.version.cuda))

可以看到,是编译过程中有函数在检查版本是否一致。
定位到 cpp_extension.py:525
本地编译安装 Minkowski Engine 报错 Cuda 版本 与 Pytorch 版本不匹配_第1张图片
然后将 _check_cuda_version 函数注释掉,然后可以顺利使用兼容的 Pytorch 进入编译的下一阶段。

你可能感兴趣的:(随笔,pytorch,人工智能,python)