Windows10 利用 Docker 配置 TensofFlow 深度学习工具

时间:2023-03-08 15:37:57

TensorFlow

这个不用多介绍了吧,大家都知道,Google的开源深度学习软件库,官网点这里:https://www.tensorflow.org/

当然这个工具官方支持装在 Ubuntu 和 Mac OS X 上,想要在 Windows 下使用,就需要借助 Docker 了,看了几篇相关的 Blog,都有点过时了。

安装步骤

  1. 获取Docker for Windows 10 Professional or Enterprise 64-bit:https://www.docker.com/ 自己安装,没啥问题
  2. (可选)安装 Kitematic ,这是 Docker 的一个图形界面,只是为了用起来方便
  3. 打开 Kitematic,在 Container 中下载 TensorFlow
    1.   Windows10 利用 Docker 配置 TensofFlow 深度学习工具
  4. 运行就能用了……超简单

注意事项

Docker 默认占用 2G 内存和2 个核(好像是,至少我的是这样),这就会影响到 TensorFlow 计算的速度,如果你对这个要求不高,就不必担心了,毕竟真到要搞深度学习也是在服务器上个吧……

这个镜像有 CPU 和 GPU 模式,详情参考官方的文档:https://hub.docker.com/r/tensorflow/tensorflow/