One - One Code All

Blog Content

TensorFlow Docker镜像下载

机器学习 容器化   2017-07-13 23:15:23

tensorflow 1.14.0 镜像制作

docker pull tensorflow/tensorflow:1.14.0-py3
 docker run -dit -v /root/models:/root/models --name bert tensorflow/tensorflow:1.14.0-py3

官方 TensorFlow Docker 映像位于 tensorflow/tensorflow Docker Hub 代码库中。映像版本按照以下格式进行标记

标记说明
latestTensorFlow CPU 二进制映像的最新版本。(默认版本)
nightlyTensorFlow 映像的每夜版(不稳定)。
version指定 TensorFlow 二进制映像的版本,例如:2.1.0
develTensorFlow master 开发环境的每夜版。包含 TensorFlow 源代码。

每个基本标记都有用于添加或更改功能的变体:

标记变体说明
tag-gpu支持 GPU 的指定标记版本。详见下文
tag-py3支持 Python 3 的指定标记版本。
tag-jupyter带有 Jupyter 的指定标记版本(包含 TensorFlow 教程笔记本)

您可以一次使用多个变体。例如,以下命令会将 TensorFlow 版本映像下载到计算机上:

docker pull tensorflow/tensorflow                     # latest stable release
docker pull tensorflow/tensorflow:devel-gpu           # nightly dev release w/ GPU support
docker pull tensorflow/tensorflow:latest-gpu-jupyter  # latest release w/ GPU support and Jupyter


启动 TensorFlow Docker 容器

要启动配置 TensorFlow 的容器,请使用以下命令格式:

docker run [-it] [--rm] [-p hostPort:containerPort] tensorflow/tensorflow[:tag] [command]


有关详情,请参阅 Docker 运行参考文档

使用仅支持 CPU 的映像的示例

我们使用带 latest 标记的映像验证 TensorFlow 安装效果。Docker 会在首次运行时下载新的 TensorFlow 映像:

docker run -it --rm tensorflow/tensorflow \
   python
-c "import tensorflow as tf; print(tf.reduce_sum(tf.random.normal([1000, 1000])))"



参考连接:

https://tensorflow.google.cn/install/docker?hl=zh-cn

https://hub.docker.com/r/tensorflow/tensorflow/



上一篇:pandas写入读取h5文件
下一篇:python执行命令subprocess,Popen,PIPE

The minute you think of giving up, think of the reason why you held on so long.