Setup tensorflow in Docker

最新推荐文章于 2024-09-04 16:03:59 发布

小白笑苍

最新推荐文章于 2024-09-04 16:03:59 发布

阅读量221

点赞数

分类专栏： ML 文章标签： tensorflow docker

本文链接：https://blog.csdn.net/toyijiu/article/details/84567850

版权

ML 专栏收录该内容

14 篇文章 0 订阅

订阅专栏

you can find the info here:PUBLIC REPOSITORY tensorflow/tensorflow

pull it:

docker pull tensorflow/tensorflow:latest

Start CPU only container

$ docker run -it -p 8888:8888 tensorflow/tensorflow

Go to your browser on http://localhost:8888/

question1
When I enter this web page, it command me to input the password but I do not set it before. So I need to Setup a Password:

But where is the token value? you can find it in the shell log, just like:
question2
in the jupyter I find that it only supports the python2 but I want the python3 kernel, so how to install it?
start a new terminal, with the command python3 --version I can find the python3 has been installed, but the pip3 is not by the command python3 -m pip --version.
Firstly install pip3,search the linux version with comand cat /proc/version,find it is linuxkit,then:

apt-get udpate
apt-get install python3-pip
python3 -m pip install ipykernel
python3 -m ipykernel install --user

but there is something wrong with my network and I can not find a way to make the python2 and python3 exist together, so I have to setup tensorflow again with the python3:

docker pull tensorflow/tensorflow:latest-py3
docker run -it -p 8888:8888 tensorflow/tensorflow:latest-py3

在这里插入图片描述

Start GPU (CUDA) container

Install nvidia-docker and run:

$ docker run --runtime=nvidia -it -p 8888:8888 tensorflow/tensorflow:latest-gpu

then Go to your browser on http://localhost:8888/

Other versions (like release candidates, nightlies and more)

See the list of tags. Devel docker images include all the necessary dependencies to build from source whereas the other binaries simply have TensorFlow installed.