安装Bazel

TensorFlow编译时需要使用Bazel工具。TensorFlow 2.13.0依赖的Bazel版本为5.4.0,请下载正确的Bazel版本。

  1. 使用PuTTY工具,以root用户登录服务器。
  2. 下载Bazel 5.4.0源码至服务器。

    下载地址:https://github.com/bazelbuild/bazel/releases/download/5.4.0/bazel-5.4.0-dist.zip

    若服务器可以访问外网,则可以直接使用wget命令下载。

    1
    wget https://github.com/bazelbuild/bazel/releases/download/5.4.0/bazel-5.4.0-dist.zip --no-check-certificate
    

  3. 执行以下命令安装Bazel依赖。

    1
    yum install java-11-openjdk java-11-openjdk-devel
    

    检查OpenJDK 11是否安装成功。

    1
    java -version
    

    回显显示上述信息,表示安装成功。

    若OpenJDK版本低于此版本,编译Bazel可能会发生错误,导致无法正常构建Bazel。

  4. 可选: 若执行上述命令发现版本不匹配,请配置环境变量。

    1. 查找安装目录。
      1
      find / -name java
      
    2. 找到安装目录后,配置环境变量,假设安装目录为“/usr/lib/jvm/java-11-openjdk-11.0.21.9-1.oe2203sp3.aarch64”。
      • 方法一(永久生效):
        1
        2
        3
        echo "export JAVA_HOME=/usr/lib/jvm/java-11-openjdk-11.0.21.9-1.oe2203sp3.aarch64" >> /etc/profile
        echo "export PATH=\${JAVA_HOME}/bin:\${PATH}" >> /etc/profile
        source /etc/profile
        
      • 方法二(当前会话终端临时生效):
        1
        2
        export JAVA_HOME=/usr/lib/jvm/java-11-openjdk-11.0.21.9-1.oe2203sp3.aarch64
        export PATH=/usr/lib/jvm/java-11-openjdk-11.0.21.9-1.oe2203sp3.aarch64/bin:$PATH
        
    3. 检查OpenJDK版本信息。
      1
      java -version
      

  5. 安装解压并编译Bazel。

    1
    2
    3
    unzip bazel-5.4.0-dist.zip -d bazel
    cd bazel
    env EXTRA_BAZEL_ARGS="--tool_java_runtime_version=local_jdk" bash ./compile.sh
    

  6. 建立软链接。

    1
    ln -s ${bazel根目录}/output/bazel /usr/bin/bazel
    

    ${bazel根目录}以实际路径为准,假设Baszel源码路径为“/path/to/bazel”,则上述命令为ln -s /path/to/bazel/output/bazel /usr/bin/bazel

  7. 验证安装是否成功。

    1
    bazel --version
    

    回显显示对应的版本信息,表示安装成功。