tensorflow gpu python3.5_从源代码构建

从源代码构建 TensorFlow pip 软件包并将其安装在 Ubuntu Linux 和 macOS 上。尽管这些说明可能适用于其他系统,但仅针对 Ubuntu 和 macOS 进行了测试并在这两种平台上受支持。

注意:我们已针对 Linux 和 macOS 系统提供了经过充分测试的预构建 TensorFlow 软件包。

Linux 和 macOS 设置

安装以下构建工具以配置开发环境。

安装 Python 和 TensorFlow 软件包依赖项

Ubuntu

sudo apt install python3-dev python3-pip

macOS

需要 Xcode 9.2 或更高版本。

使用 Homebrew 软件包管理器安装:

/usr/bin/ruby -e "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/master/install)"

export PATH="/usr/local/opt/python/libexec/bin:$PATH"

# if you are on macOS 10.12 (Sierra) use export PATH="/usr/local/bin:/usr/local/sbin:$PATH"

brew install python

安装 TensorFlow pip 软件包依赖项(如果使用虚拟环境,请省略 --user 参数):

pip install -U --user pip numpy wheel

pip install -U --user keras_preprocessing --no-deps

注意:必须使用 pip 19.0 以上的版本才能安装 TensorFlow 2 .whl 软件包。请参见 setup.py 文件中的 REQUIRED_PACKAGES 部分,了解其他必需的依赖项。

安装 Bazel

您需要安装 Bazel,才能构建 TensorFlow。您可以使用 Bazelisk 轻松安装 Bazel,并且 Bazelisk 可以自动为 TensorFlow 下载合适的 Bazel 版本。为便于使用,请在 PATH 中将 Bazelisk 添加为 bazel 可执行文件。

如果没有 Bazelisk,您可以手动安装 Bazel。请务必安装受支持的 Bazel 版本,可以是 tensorflow/configure.py 中指定的介于 _TF_MIN_BAZEL_VERSION 和 _TF_MAX_BAZEL_VERSION 之间的任意版本。

安装 GPU 支持项(可选,仅限 Linux)

没有针对 macOS 的 GPU 支持版本。

如需安装在 GPU 上运行 TensorFlow 所需的驱动程序和其他软件,请参阅 GPU 支持指南。

注意:您可以轻松设置 TensorFlow 的某个支持 GPU 的 Docker 映像。

下载 TensorFlow 源代码

git clone https://github.com/tensorflow/tensorflow.git

cd tensorflow

该代码库默认为 master 开发分支。您也可以检出要构建的版本分支:

git checkout branch_name # r2.2, r2.3, etc.

配置 build

通过运行 TensorFlow 源代码树根目录下的 ./configure 配置系统 build。此脚本会提示您指定 TensorFlow 依赖项的位置,并要求指定其他构建配置选项(例如,编译器标记)。

./configure

如果使用的是虚拟环境,python configure.py 会优先处理环境内的路径,而 ./configure 会优先处理环境外的路径。在这两种情况下,您都可以更改默认设置。

示例会话

以下代码展示了 ./configure 脚本的示例运行会话(您的会话可能会有所不同):

查看示例配置会话

./configure

You have bazel 3.0.0 installed.

Please specify the location of python. [Default is /usr/bin/python3]:

Found possible Python library paths:

/usr/lib/python3/dist-packages

/usr/local/lib/python3.6/dist-packages

Please input the desired Python library path to use. Default is [/usr/lib/python3/dist-packages]

Do you wish to build TensorFlow with OpenCL SYCL support? [y/N]:

No OpenCL SYCL support will be enabled for TensorFlow.

Do you wish to build TensorFlow with ROCm support? [y/N]:

No ROCm 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 wish to build TensorFlow with TensorRT support? [y/N]:

No TensorRT support will be enabled for

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
C:\Users\zhang'xin'ge>python -c "import tensorflow as tf; print(tf._version_)" Traceback (most recent call last): File "<string>", line 1, in <module> File "D:\Anaconda\lib\site-packages\tensorflow\__init__.py", line 38, in <module> from tensorflow.python.tools import module_util as _module_util File "D:\Anaconda\lib\site-packages\tensorflow\python\__init__.py", line 37, in <module> from tensorflow.python.eager import context File "D:\Anaconda\lib\site-packages\tensorflow\python\eager\context.py", line 32, in <module> from tensorflow.python import pywrap_tfe File "D:\Anaconda\lib\site-packages\tensorflow\python\pywrap_tfe.py", line 25, in <module> from tensorflow.python._pywrap_tfe import * ImportError: DLL load failed while importing _pywrap_tfe: 找不到指定的模块。 C:\Users\zhang'xin'ge>python -c "import keras; print(keras._version_)" Traceback (most recent call last): File "<string>", line 1, in <module> File "D:\Anaconda\lib\site-packages\keras\__init__.py", line 21, in <module> from tensorflow.python import tf2 File "D:\Anaconda\lib\site-packages\tensorflow\__init__.py", line 38, in <module> from tensorflow.python.tools import module_util as _module_util File "D:\Anaconda\lib\site-packages\tensorflow\python\__init__.py", line 37, in <module> from tensorflow.python.eager import context File "D:\Anaconda\lib\site-packages\tensorflow\python\eager\context.py", line 32, in <module> from tensorflow.python import pywrap_tfe File "D:\Anaconda\lib\site-packages\tensorflow\python\pywrap_tfe.py", line 25, in <module> from tensorflow.python._pywrap_tfe import * ImportError: DLL load failed while importing _pywrap_tfe: 找不到指定的模块。
07-23

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值