热线电话:13121318867

登录
2018-12-01 阅读量: 844
使用 Docker 安装tensorflow

按照以下步骤使用 Docker 安装 TensorFlow:

按照 文档 在你的机器上安装 Docker

启动任一个包含 TensorFlow 镜像的 Docker 容器

本节剩下部分解释如何启动 Docker 容器。

要启动包含 TensorFlow 镜像的 Docker 容器,执行以下命令:

 $ docker run -it -p hostPort:containerPort TensorFlowImage 

where:

-p hostPort:containerPort 是可选的,如果你想从 shell 运行 TensorFlow 程序忽略这个选项。如果你想从 Jupyter notebook 运行 TensorFlow 程序,hostPort 和 containerPort 都设置为 8888。如果你想在镜像中运行 TensorBoard,再添加一个-p 参数,hostPort 和 containerPort 都设置为 6006。

TensorFlowImage 是需要的,它用于指定 Docker 容器,你必须指定接下来的任一一个:gcr.io/tensorflow/tensorflow: TensorFlow 二进制镜像,gcr.io/tensorflow/tensorflow:latest-devel: TensorFlow 二进制镜像加源码。

gcr.io 是 Goole 的容器注册表,注意部分 TensorFlow 也可以从 dockerhub 获取。

例如,下面的命令可以在 Docker 容器中启动一个 TensorFlow CPU 镜像,然后你可以在镜像的 shell 中运行 TensorFlow 程序:r run -it -p hostPort:containerPort TensorFlowIma

$ docker run -it gcr.io/tensorflow/tensorflow bashge 

以下命令也可以在 Docker 容器中启动一个 TensorFlow CPU 镜像,然而,在这个 Docker 镜像中,你可以在 Jupyter notebook 中运行 TensorFlow 程序:

$ docker run -it -p 8888:8888 gcr.io/tensorflow/tensorflow
0.0000
0
关注作者
收藏
评论(0)

发表评论

暂无数据
推荐帖子