报错详情

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：

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