docker安装TensorFlow

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

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

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

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

Original: https://blog.csdn.net/yinjl123456/article/details/124052347
Author: beyond阿亮
Title: docker安装TensorFlow

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/496903/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球