Ubuntu 基于Docker的TensorFlow 环境搭建

时间:2022-09-16 15:36:55

基于Docker的TensorFlow 环境搭建

基于(ubuntu 16.04LTS/ubuntu 14.04LTS)

一、docker环境安装

1)更新、安装依赖包

sudo apt-get update
sudo apt-get install
apt-transport-https
ca-certificates
curl
software-properties-common

2)添加官方密钥

curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo apt-key add -

3)添加仓库

sudo add-apt-repository
"deb [arch=amd64] https://download.docker.com/linux/ubuntu
$(lsb_release -cs)
stable"

4)安装docker

A:更新安装包

sudo apt-get update

安装最新版

sudo apt-get install lxc-docker

安装指定版

sudo apt-get install docker-ce

查看版本

docker -v

二、TensorFlow 环境搭建

1)下载TensorFlow镜像

docker pull tensorflow/tensorflow
或者
sudo docker pull tensorflow/tensorflow

2)创建运行TensorFlow容器

docker run --name my-tensortflow -it -p 8888:8888 -v ~/tensorflow:/test/data tensorflow/tensorflow

docker run运行镜像:

--name 为创建容器名 即:my-tensortflow

-it 保留命令行运行

-p 8888:8888 将本地的8888端口http://localhost:8888/映射,

-v ~/tensorflow:/notebooks/data 将本地的~/tensorflow文件夹挂载到新建容器的/test/data下(这样创建的文件可以保存到本地~/tensorflow)
tensorflow/tensorflow为指定的镜像,默认标签为latest(即tensorflow/tensorflow:latest)

3)开启TensorFlow容器

A: 在浏览器中输入localhost:8888
B: 将命令行生成的token 贴在网页的passwor or token 框里,点login按钮
C:在首页可以新建一个python 来测试

关闭TensorFlow和开启TensorFlow环境

4)开启、关闭tensorflow容器

关闭:

docker stop my-tensortflow

开启

docker start my-tensortflow
浏览器中输入 http://localhost:8888/ 登录