Docker环境下PIP安装Horovod失败：C++17编译器配置问题

核心原因 #

你之前使用的MMVC_CUDA_ARGS和HOROVOD_BUILD_ARCH_FLAGS并非Horovod识别的标准编译参数环境变量，导致CMake编译时仍采用默认的C++14标准，触发了ATen的版本要求错误。

解决方案 #

1. 使用正确的环境变量执行安装命令 #

替换原有安装命令，使用Horovod官方支持的编译参数变量，或全局编译器变量强制指定C++17：

方法一：使用Horovod专属变量

HOROVOD_CXX_FLAGS="-std=c++17" HOROVOD_CUDA_CXX_FLAGS="-std=c++17" HOROVOD_WITHOUT_MXNET=1 HOROVOD_WITHOUT_TENSORFLOW=1 pip install horovod[pytorch]

方法二：设置全局编译器变量

CXXFLAGS="-std=c++17" CUDAFLAGS="-std=c++17" HOROVOD_WITHOUT_MXNET=1 HOROVOD_WITHOUT_TENSORFLOW=1 pip install horovod[pytorch]

两种方式都会强制编译器在编译Horovod的C代码时使用C17标准，覆盖默认的C++14设置。

2. 将Horovod安装整合到Dockerfile（推荐） #

避免手动进入容器操作，直接在Dockerfile中添加安装步骤，确保环境变量全程生效：

修改后的Dockerfile片段：

RUN ${PYTHON_BIN} -m pip install torch torchvision pyyaml pandas scikit-image openexr

# 设置C++17编译参数
ENV HOROVOD_CXX_FLAGS="-std=c++17"
ENV HOROVOD_CUDA_CXX_FLAGS="-std=c++17"

# 安装Horovod
RUN HOROVOD_WITHOUT_MXNET=1 HOROVOD_WITHOUT_TENSORFLOW=1 ${PYTHON_BIN} -m pip install horovod[pytorch]

额外提示 #

你的Dockerfile中PYTHON_SITE_PACKAGES变量存在语法错误，需将{PYTHON_BIN}改为${PYTHON_BIN}：

ENV PYTHON_SITE_PACKAGES="/usr/local/lib/${PYTHON_BIN}/dist-packages"

验证 #

重新构建镜像或执行安装命令后，查看编译日志中的g++命令，确认参数中包含-std=c++17而非-std=c++14，即可解决ATen的编译错误。

内容的提问来源于stack exchange，提问作者Blueshark