从源代码构建 TensorFlow 安装

悠悠 2023-03-12 09:26 7阅读 0赞

#  从源代码构建 TensorFlow 安装 #

## 配置 build ##

通过运行 TensorFlow 源代码树根目录下的 `./configure` 配置系统 build。此脚本会提示您指定 TensorFlow 依赖项的位置，并要求指定其他构建配置选项。

./configure

以下 `./configure` 脚本展示了 `./configure` 的示例运行会话，查看示例配置会话。

You have bazel 0.15.0 installed.
        Please specify the location of python. [Default is /usr/bin/python]: /usr/bin/python2.7
    
        Found possible Python library paths:
          /usr/local/lib/python2.7/dist-packages
          /usr/lib/python2.7/dist-packages
        Please input the desired Python library path to use.  Default is [/usr/lib/python2.7/dist-packages]
    
        Do you wish to build TensorFlow with jemalloc as malloc support? [Y/n]:
        jemalloc as malloc support will be enabled for TensorFlow.
    
        Do you wish to build TensorFlow with Google Cloud Platform support? [Y/n]: -> n
        Google Cloud Platform support will be enabled for TensorFlow.
    
        Do you wish to build TensorFlow with Hadoop File System support? [Y/n]: -> n
        Hadoop File System support will be enabled for TensorFlow.
    
        Do you wish to build TensorFlow with Amazon AWS Platform support? [Y/n]: -> n
        Amazon AWS Platform support will be enabled for TensorFlow.
    
        Do you wish to build TensorFlow with Apache Kafka Platform support? [Y/n]:
        Apache Kafka Platform support will be enabled for TensorFlow.
    
        Do you wish to build TensorFlow with XLA JIT support? [y/N]:
        No XLA JIT support will be enabled for TensorFlow.
    
        Do you wish to build TensorFlow with GDR support? [y/N]:
        No GDR support will be enabled for TensorFlow.
    
        Do you wish to build TensorFlow with VERBS support? [y/N]:
        No VERBS support will be enabled for TensorFlow.
    
        Do you wish to build TensorFlow with OpenCL SYCL support? [y/N]: -> n
        No OpenCL SYCL support will be enabled for TensorFlow.
    
        Do you wish to build TensorFlow with CUDA support? [y/N]: -> Y
        CUDA support will be enabled for TensorFlow.
    
        Please specify the CUDA SDK version you want to use. [Leave empty to default to CUDA 9.0]: -> 9.0
    
        Please specify the location where CUDA 9.0 toolkit is installed. Refer to README.md for more details. [Default is /usr/local/cuda]:
    
        Please specify the cuDNN version you want to use. [Leave empty to default to cuDNN 7.0]: -> 7.0
    
        Please specify the location where cuDNN 7 library is installed. Refer to README.md for more details. [Default is /usr/local/cuda]:
    
        Do you wish to build TensorFlow with TensorRT support? [y/N]:
        No TensorRT support will be enabled for TensorFlow.
    
        Please specify the NCCL version you want to use. If NCLL 2.2 is not installed, then you can use version 1.3 that can be fetched automatically but it may have worse performance with multiple GPUs. [Default is 2.2]: 1.3
    
        Please specify a list of comma-separated Cuda compute capabilities you want to build with.
        You can find the compute capability of your device at: https://developer.nvidia.com/cuda-gpus.
        Please note that each additional compute capability significantly increases your
        build time and binary size. [Default is: 3.5,7.0] 6.1
    
        Do you want to use clang as CUDA compiler? [y/N]:
        nvcc will be used as CUDA compiler.
    
        Please specify which gcc should be used by nvcc as the host compiler. [Default is /usr/bin/gcc]:
    
        Do you wish to build TensorFlow with MPI support? [y/N]:
        No MPI support will be enabled for TensorFlow.
    
        Please specify optimization flags to use during compilation when bazel option "--config=opt" is specified [Default is -march=native]:
    
        Would you like to interactively configure ./WORKSPACE for Android builds? [y/N]:
        Not configuring the WORKSPACE for Android builds.
    
        Preconfigured Bazel build configs. You can use any of the below by adding "--config=<>" to your build command. See tools/bazel.rc for more details.
            --config=mkl            # Build with MKL support.
            --config=monolithic     # Config for mostly static monolithic build.
        Configuration finished