概述:基于Docker的TensorFlow机器学习框架搭建和实例源码解读,TensorFlow做为最火热的机器学习框架之一,Docker是的容器,能够很好的结合起来,为机器学习或者科研人员提供便捷的机器学习开发环境,探索人工智能的奥秘,容器随开随用方便快捷。源码解析TensorFlow容器建立和示例程序运行,为热爱机器学者下降学习难度。python
默认机器已经装好了Docker(Docker安装和使用能够看我另外一篇博文:Ubuntu16.04安装Docker1.12+开发实例+hello world+web应用容器)。linux
docker pull tensorflow/tensorflow
#或者
#sudo docker pull tensorflow/tensorflow
docker run --name xiaolei-tensortflow -it -p 8888:8888 -v ~/tensorflow:/notebooks/data tensorflow/tensorflow
docker run
运行镜像,--name
为容器建立别名,-it
保留命令行运行,-p 8888:8888
将本地的8888端口http://localhost:8888/
映射,-v ~/tensorflow:/notebooks/data
将本地的~/tensorflow文件夹挂载到新建容器的/notebooks/data下(这样建立的文件能够保存到本地~/tensorflow)tensorflow/tensorflow
为指定的镜像,默认标签为latest(即tensorflow/tensorflow:latest)New
一个Python项目from __future__ import print_function
#导入tensorflow
import tensorflow as tf
#输入两个数组,input1和input2而后相加,输出结果
with tf.Session():
input1 = tf.constant([1.0, 1.0, 1.0, 1.0])
input2 = tf.constant([2.0, 2.0, 2.0, 2.0])
output = tf.add(input1, input2)
result = output.eval()
print("result: ", result)
result: [ 3. 3. 3. 3.]
#关闭tensorflow容器
docker stop xiaolei-tensortflow
#开启TensorFlow容器
docker start xiaolei-tensortflow
#浏览器中输入 http://localhost:8888/
#查看读写权限
ls -l
#将tensorflow 变为属于xiaolei(系统默认)用户
sudo chown -R xiaolei tensorflow/
#将tensorflow 变为属于xiaolei(系统默认)用户组
sudo chgrp -R xiaolei tensorflow/