构建机器学习Docker容器时TensorFlow编译问题求助

问题清单 #

• Bazel安装版本收到非正式版警告
• ./configure 无法通过交互式提示完成配置（Docker构建为无交互环境）
• 编译报错：ERROR: The project you're trying to build requires Bazel 5.3.0 (specified in /docklearning/tensorflow/.bazelversion), but it wasn't found in /usr/bin

针对性修复步骤 #

1. 安装指定版本Bazel（解决版本不匹配+非正式版警告）

替换原Dockerfile中Bazel的apt安装逻辑，直接下载TensorFlow要求的5.3.0官方正式二进制文件，避免apt源版本不符的问题：

# 移除原Bazel apt源配置及apt install bazel命令，替换为以下代码
RUN wget https://github.com/bazelbuild/bazel/releases/download/5.3.0/bazel-5.3.0-linux-x86_64 && \
    chmod +x bazel-5.3.0-linux-x86_64 && \
    mv bazel-5.3.0-linux-x86_64 /usr/bin/bazel

2. 非交互式完成TensorFlow配置

Docker构建过程无法响应交互式输入，需通过设置环境变量+自动输入流完成./configure：

# 在git clone tensorflow后、cd tensorflow的步骤中添加配置逻辑
RUN git clone https://github.com/tensorflow/tensorflow.git && \
    cd tensorflow && \
    # 根据硬件和需求设置自动配置环境变量（示例适配CUDA 12.0环境）
    export TF_NEED_CUDA=1 && \
    export TF_CUDA_VERSION=12.0 && \
    export TF_CUDNN_VERSION=8 && \
    export TF_CUDA_COMPUTE_CAPABILITIES="7.5,8.0,8.6" && \
    export PYTHON_BIN_PATH=/opt/conda/bin/python && \
    export CC_OPT_FLAGS="-mavx2" && \
    # 用EOF自动完成交互式输入
    ./configure <<EOF
EOF
    # 后续编译命令保持不变
    bazel build --config=opt --config=cuda --cxxopt="-mavx2" //tensorflow/tools/pip_package:build_pip_package && \
    ./bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg && \
    pip install /tmp/tensorflow_pkg/tensorflow-*.whl && \
    cd .. && \
    rm -rf tensorflow

注：TF_CUDA_COMPUTE_CAPABILITIES需匹配你的GPU算力，可根据实际硬件调整。

3. 额外优化建议

• 建议克隆指定版本的TensorFlow分支，避免最新分支依赖的Bazel版本变更：

  # 示例：克隆适配Bazel 5.3.0的r2.13分支
  RUN git clone --branch r2.13 https://github.com/tensorflow/tensorflow.git
  

• 确认conda环境的Python版本符合TensorFlow要求（r2.13版本推荐Python 3.8-3.11）

修改后的关键Dockerfile片段 #

# Install packages
RUN apt-get update -q && apt-get install -q -y --no-install-recommends \
    apt-transport-https curl gnupg apt-utils wget gcc g++ npm unzip build-essential ca-certificates curl git gh \
    make nano iproute2 nano openssh-client openssl procps \
    software-properties-common bzip2 subversion neofetch \
    fontconfig && \
    # 安装指定版本Bazel 5.3.0
    wget https://github.com/bazelbuild/bazel/releases/download/5.3.0/bazel-5.3.0-linux-x86_64 && \
    chmod +x bazel-5.3.0-linux-x86_64 && \
    mv bazel-5.3.0-linux-x86_64 /usr/bin/bazel && \
    apt-get full-upgrade -q -y && \
    cd ~ && \
    wget https://github.com/ryanoasis/nerd-fonts/releases/download/v2.1.0/Meslo.zip && \
    mkdir -p .local/share/fonts && \
    unzip Meslo.zip -d .local/share/fonts && \
    cd .local/share/fonts && rm *Windows* && \
    cd ~ && \
    rm Meslo.zip && \
    fc-cache -fv && \
    apt-get install -y zsh zsh-doc chroma

# ...（中间Anaconda安装等部分保持不变）

# Download Tensorflow from github and build from source
RUN git clone --branch r2.13 https://github.com/tensorflow/tensorflow.git && \
    cd tensorflow && \
    # 设置自动配置环境变量
    export TF_NEED_CUDA=1 && \
    export TF_CUDA_VERSION=12.0 && \
    export TF_CUDNN_VERSION=8 && \
    export TF_CUDA_COMPUTE_CAPABILITIES="7.5,8.0,8.6" && \
    export PYTHON_BIN_PATH=/opt/conda/bin/python && \
    export CC_OPT_FLAGS="-mavx2" && \
    # 非交互式执行configure
    ./configure <<EOF
EOF
    bazel build --config=opt --config=cuda --cxxopt="-mavx2" //tensorflow/tools/pip_package:build_pip_package && \
    ./bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg && \
    pip install /tmp/tensorflow_pkg/tensorflow-*.whl && \
    cd .. && \
    rm -rf tensorflow

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