QDS04 TensorFlow

关注

发布于: 刚刚

QDS （Quick Deployment Series）快速的部署一个软件。这次我们来部署 TensorFlow 。

系统要求

Ubuntu 16.04 或更高版本（64 位）macOS 10.12.6 (Sierra) 或更高版本（64 位）（不支持 GPU）Windows 7 或更高版本（64 位）（仅支持 Python 3）Raspbian 9.0 或更高版本

硬件要求

从 TensorFlow 1.6 开始，二进制文件使用 AVX 指令，这些指令可能无法在旧版 CPU 上运行。

CUDA® 计算能力为 3.5 或更高的 NVIDIA® GPU 卡。

软件要求

必须在系统中安装以下 NVIDIA® 软件：

NVIDIA® GPU 驱动程序 - CUDA 9.0 需要 384.x 或更高版本。
CUDA® 工具包 - TensorFlow 支持 CUDA 9.0。
CUDA 工具包附带的 CUPTI。
cuDNN SDK（7.2 及更高版本）
（可选）NCCL 2.2，可实现多 GPU 支持。
（可选）TensorRT 4.0，可缩短在某些模型上进行推断的延迟并提高吞吐量。

安装

环境检测

python 2 环境检测，如果没有某个部分，请补装。

python --versionpip --version

复制代码

python3 和 python2 类似，换成 python3 的命令即可。

安装

pip install tensorflow  // 安装 TensorFlow CPU 版本pip install tensorflow-gpu  // 安装 TensorFlow GPU 版本pip install tf-nightly    // 安装 TensorFlow CPU 暗夜版本 不稳定pip install tf-nightly-gpu   // 安装 TensorFlow GPU 暗夜版本 不稳定

复制代码

不过这样通过 pip 安装速度有点慢啊，使用了 pip 官方的源。

换阿里云的 pip 源以后很快就完成了。

pip install -i https://mirrors.aliyun.com/pypi/simple tensorflow-gpu

复制代码

验证

依次输入如下命令：

[root@node31 ~]# pythonPython 2.7.5 (default, Jul 13 2018, 13:06:57)[GCC 4.8.5 20150623 (Red Hat 4.8.5-28)] on linux2Type "help", "copyright", "credits" or "license" for more information.>>> import tensorflow as tf>>> tf.__version__'1.12.0'>>> tf.__path__['/usr/lib/python2.7/site-packages/tensorflow', '/usr/lib/python2.7/site-packages/tensorflow/python/estimator/api', '/usr/lib/python2.7/site-packages/tensorflow/_api/v1']>>>

复制代码

可以看到，已经安装了 TensorFlow 的 1.12 版本和安装的路径。

发布于: 刚刚阅读数: 2

原文链接:【http://xie.infoq.cn/article/a9904317c3e1538783d113164】。未经作者许可，禁止转载。

耳东

关注

还未添加个人签名 2020.05.24 加入

主要研究分享运维技术，专注于监控、CICD、操作系统、云原生领域，公众号【耳东学堂】，知识星球同名，坚持原创，希望能和大家在运维路上结伴而行邮箱：erdong@mail.erdong.site

发布

暂无评论

创作场景

QDS04 TensorFlow

系统要求

硬件要求

软件要求

安装

环境检测

安装

验证

耳东

评论