在线时间:8:00-16:00
迪恩网络APP
随时随地掌握行业动态
扫描二维码
关注迪恩网络微信公众号
本指南介绍了如何通过源代码编译安装Tesnorflow。请注意,我们已经为Linux,Mac和Windows系统提供了经充分测试并预编译好的TensorFlow二进制文件。另外还有预编译的TensorFlowDocker映像。所以,不要自己构建一个TensorFlow二进制文件,除非你乐于从源代码构建,并愿意处理一些未知的问题。 本指南介绍如何在以下操作系统上构建TensorFlow:
目前没有正式支持在Windows上构建TensorFlow;但是,如果您不介意使用高度实验性的Bazel在Windows上或TensorFlow CMake构建,可以自己尝试在Windows上构建TensorFlow。 确定安装哪个TensorFlow您必须选择以下类型的TensorFlow来构建和安装:
克隆TensorFlow存储库从克隆TensorFlow存储库开始构建TensorFlow。 要克隆最新的TensorFlow存储库,执行以下命令: $ git clone https://github.com/tensorflow/tensorflow 前面的 $ cd tensorflow $ git checkout Branch # where Branch is the desired branch 例如,要使用 $ git checkout r1.0 接下来,您必须为Linux或Mac OS准备好您的环境 为Linux准备环境在Linux上构建TensorFlow之前,请在系统上安装以下构建工具:
安装Bazel如果系统上没有安装bazel,请按照以下步骤进行安装见文档。 安装TensorFlow Python依赖项要安装TensorFlow,您必须安装以下软件包:
要为Python 2.7安装这些软件包,执行以下命令: $ sudo apt-get install python-numpy python-dev python-pip python-wheel 要为Python 3.n安装这些软件包,执行以下命令: $ sudo apt-get install python3-numpy python3-dev python3-pip python3-wheel 可选:安装TensorFlow GPU版本的要求如果您正在构建没有GPU支持的TensorFlow,请跳过本节。 以下NVIDIA硬件必须安装在您的系统上:
以下NVIDIA软件必须安装在您的系统上:
最后,您还必须安装 $ sudo apt-get install libcupti-dev 下一个在准备好环境后,你现在必须做配置安装。 为Mac OS准备环境在构建TensorFlow之前,您必须在系统上安装以下内容:
安装bazel如果系统上没有安装bazel,请按照以下步骤进行安装,见文档。 安装python依赖项要安装TensorFlow,您必须安装以下软件包:
您可以使用pip安装python依赖项。如果您的机器上没有pip,我们建议使用homebrew来安装Python和pip,见这个文档。如果您遵循这些说明,则不需要禁用SIP。 安装pip后,调用以下命令: $ sudo pip install six numpy wheel 可选:安装TensorFlow GPU版本的要求如果没有安装brew,请按照以下步骤进行安装,见brew安装说明。 安装brew后,执行以下命令安装GNU coreutils: $ brew install coreutils 如果要编译Tensorflow并安装XCode 7.3和CUDA 7.5,请注意,Xcode 7.3尚未兼容CUDA 7.5。要解决此问题,请执行以下任一操作:
注意:您的系统必须满足以下文档描述的NVIDIA软件要求:
配置安装源代码树的根目录包含一个名为 如果你想构建支持GPU的TensorFlow, 其中一个 Please specify optimization flags to use during compilation when bazel option "--config=opt" is specified [Default is -march=native] 这个问题出现在构建pip包时。我们建议接受默认值( 下面执行 $ cd tensorflow # cd to the top-level directory created $ ./configure 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] Using python library path: /usr/local/lib/python2.7/dist-packages Do you wish to build TensorFlow with MKL support? [y/N] No MKL 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]: Do you wish to use jemalloc as the malloc implementation? [Y/n] jemalloc enabled Do you wish to build TensorFlow with Google Cloud Platform support? [y/N] No Google Cloud Platform support will be enabled for TensorFlow Do you wish to build TensorFlow with Hadoop File System support? [y/N] No Hadoop File System support will be enabled for TensorFlow Do you wish to build TensorFlow with the XLA just-in-time compiler (experimental)? [y/N] No XLA 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 support? [y/N] No OpenCL 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 Do you want to use clang as CUDA compiler? [y/N] nvcc will be used as CUDA compiler Please specify the Cuda SDK version you want to use, e.g. 7.0. [Leave empty to default to CUDA 8.0]: 8.0 Please specify the location where CUDA 8.0 toolkit is installed. Refer to README.md for more details. [Default is /usr/local/cuda]: Please specify which gcc should be used by nvcc as the host compiler. [Default is /usr/bin/gcc]: Please specify the cuDNN version you want to use. [Leave empty to default to cuDNN 6.0]: 6 Please specify the location where cuDNN 6 library is installed. Refer to README.md for more details. [Default is /usr/local/cuda]: 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,5.2"]: 3.0 Do you wish to build TensorFlow with MPI support? [y/N] MPI support will not be enabled for TensorFlow Configuration finished 如果要构建支持GPU的Tensorflow, 另外,请注意以下事项:
构建pip包要为支持CPU-only的TensorFlow构建一个pip包,需要执行以下命令: $ bazel build --config=opt //tensorflow/tools/pip_package:build_pip_package 要为具有GPU支持的TensorFlow构建pip软件包,请调用以下命令: $ bazel build --config=opt --config=cuda //tensorflow/tools/pip_package:build_pip_package 关于gcc 5或更高版本的注意事项:TensorFlow网站上提供的二进制pip包是使用gcc 4编译的,它使用较旧的ABI。要使您的构建与较旧的ABI兼容,您需要添加 Tips:默认情况下,从源代码构建TensorFlow消耗大量的RAM。如果RAM有限,您可以在使用 该 $ bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg 安装pip包调用 在Linux上对于TensorFlow 1.3.0版本: $ sudo pip install /tmp/tensorflow_pkg/tensorflow-1.3.0-py2-none-any.whl 验证您的安装通过执行以下操作来验证您的TensorFlow安装: 启动一个终端 更改目录( 调用python: $ python 在python交互式shell中输入以下短程序:
如果系统输出以下内容,则可以开始编写TensorFlow程序: Hello, TensorFlow! 如果您是TensorFlow新手,请参阅TensorFlow入门。 如果系统输出错误消息而不是问候,请参阅常见的安装问题。 常见的安装问题遇到的安装问题通常取决于操作系统。请参阅以下指南之一的“常见安装问题”部分: 除了这两个指南中记录的错误之外,下表列出了构建TensorFlow特有的其他错误。
本文整理自
|
2023-10-27
2022-08-15
2022-08-17
2022-09-23
2022-08-13