站点图标 AI技术聚合

docker安装TensorFlow

docker安装TensorFlow

TensorFlow介绍

TensorFlow 是一个端到端开源机器学习平台。它拥有一个全面而灵活的生态系统,其中包含各种工具、库和社区资源,可助力研究人员推动先进机器学习技术的发展,并使开发者能够轻松地构建和部署由机器学习提供支持的应用。

TensorFlow是一个基于数据流编程(dataflow programming)的符号数学系统,被广泛应用于各类机器学习(machine learning)算法的编程实现,其前身是谷歌的神经网络算法库DistBelief。

Tensorflow拥有多层级结构,可部署于各类服务器、PC终端和网页并支持GPU和TPU高性能数值计算,被广泛应用于谷歌内部的产品开发和各领域的科学研究

docker安装TensorFlow

docker run -it --rm tensorflow/tensorflow bash

进入shell后输入 Python 在python交互式shell 下输入下面代码
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))
输出如下内容
Hello, TensorFlow!

Docker 容器中启动一个 TensorFlow CPU 二进制映像, CPU电脑下运行tensorflow
docker run -it --rm -p 8888:8888 tensorflow/tensorflow:latest-py3-jupyter

访问: http://localhost:8888/
复制token登录, token在日志中有打印出来

可以在 Jupyter Notebook 中运行 TensorFlow 程序
点击 New 选择Python3 打开 Jupyter Notebook
复制上面的代码,点击运行
在Jupyter Notebook 中正常运行了上面的代码

参考链接:
https://segmentfault.com/a/1190000015053704?utm_source=channel-hottest

文章出处登录后可见!

已经登录?立即刷新
退出移动版