在Docker容器中运行TensorFlow
环境需求
- Docker
启动步骤
1. 拉取TensorFlow镜像
$ docker pull tensorflow/tensorflow:nightly-jupyter
我这里采用的是nightly版本并且带有jupyter的环境,如果你想使用稳定版本可以把后面的镜像tag替换为latest-py3-jupyter, 即最新发行版+python3+jupyter的环境,根据需求灵活挑选。
2. 运行该镜像
$ docker run -it --rm -p 8888:8888 -v $PWD:/tf/notebooks tensorflow/tensorflow:nightly-jupyter
- -it => 以交互式方式运行
- -p 8888:8888 => 把容器的8888端口映射到物理机的8888端口
- -v $PWD:/tf/notebooks => 把物理机当前目录映射到容器的/tf/notebooks, 这样就可以把代码文件保存到物理机上。
我们从截图上可以看到要访问jupyter notebook的话就复制这个链接到浏览器:http://127.0.0.1:8888/?token=a39682c07f1275a4fe1f16cd4f635bb632c714b380c1b411
每次启动都是不一样的token,不要复制我的链接去访问。