NVIDIA/apex报错解决过程

最新推荐文章于 2026-04-11 11:02:27 发布

原创

最新推荐文章于 2026-04-11 11:02:27 发布 · 6.5k 阅读

标签

#cuda #linux

收录于

本文详细记录了解决在安装NVIDIA Apex时遇到的CUDA版本不匹配问题，包括检查和设置正确的CUDA_HOME环境变量，确保与Pytorch二进制文件的CUDA版本一致，以及处理Microsoft Visual Studio版本不支持的问题。

https://github.com/NVIDIA/apex

python setup.py install --cuda_ext --cpp_ext

马上报错

File "setup.py", line 106, in <module>
    check_cuda_torch_binary_vs_bare_metal(torch.utils.cpp_extension.CUDA_HOME)
  File "setup.py", line 80, in check_cuda_torch_binary_vs_bare_metal
    "https://github.com/NVIDIA/apex/pull/323#discussion_r287021798.  "

RuntimeError: Cuda extensions are being compiled with a version of Cuda that does not match the version used to compile Pytorch binaries.  Pytorch binaries were compiled with Cuda 10.0.
In some cases, a minor-version mismatch will not cause later errors:  https://github.com/NVIDIA/apex/pull/323#discussion_r287021798.  You can try commenting out this check (at your own risk).

Cuda扩展的编译版本与用于编译Pytorch二进制文件的版本不匹配。Pytorch二进制文件是用Cuda 10.0编译的。

查询一下CUDA版本