在Docker容器中运行TensorFlow

环境需求

  1. Docker

启动步骤

1. 拉取TensorFlow镜像

TensorFlow镜像地址

$ docker pull tensorflow/tensorflow:nightly-jupyter

拉取TensorFlow的docker镜像

我这里采用的是nightly版本并且带有jupyter的环境,如果你想使用稳定版本可以把后面的镜像tag替换为latest-py3-jupyter, 即最新发行版+python3+jupyter的环境,根据需求灵活挑选。

2. 运行该镜像

$ docker run -it --rm -p 8888:8888 -v $PWD:/tf/notebooks tensorflow/tensorflow:nightly-jupyter

运行TensorFlow的docker镜像

  • -it => 以交互式方式运行
  • -p 8888:8888 => 把容器的8888端口映射到物理机的8888端口
  • -v $PWD:/tf/notebooks => 把物理机当前目录映射到容器的/tf/notebooks, 这样就可以把代码文件保存到物理机上。

我们从截图上可以看到要访问jupyter notebook的话就复制这个链接到浏览器:http://127.0.0.1:8888/?token=a39682c07f1275a4fe1f16cd4f635bb632c714b380c1b411

每次启动都是不一样的token,不要复制我的链接去访问。

jupyter启动图

加载评论框需要科学上网