如何利用Docker运行容器化的TensorFlow深度学习环境

轻萱同学_7684

轻萱同学_7684

2026-04-27

542人浏览

原创

核心就三步:装好Docker和NVIDIA Container Toolkit、拉取tensorflow/tensorflow:2.13.0-gpu等官方镜像、运行docker run --gpus all -p 8888:8888 -v $(pwd)/notebooks:/tf/notebooks镜像启动Jupyter环境,内置Python、CUDA、cuDNN及TensorBoard,无需手动配置。

如何利用docker运行容器化的tensorflow深度学习环境

直接用 Docker 运行 TensorFlow 环境,核心就三步:装好 Docker 和 GPU 支持工具、拉取合适镜像、启动容器并验证。不需要手动装 CUDA、cuDNN 或 Python 包,官方镜像已全部预配好。

确保宿主机基础环境就绪

先确认系统已安装 Docker 引擎(≥19.03),执行 docker --version 查看版本。若未安装,Ubuntu/Debian 用户可运行:

  • sudo apt-get update && sudo apt-get install docker-ce docker-ce-cli containerd.io
  • sudo systemctl enable --now docker

如需 GPU 加速,必须额外安装 NVIDIA Container Toolkit(不是旧版 nvidia-docker2)。关键步骤包括:

  • 宿主机已装好匹配的 NVIDIA 驱动(建议 ≥535 版本)
  • 添加 NVIDIA 软件源并安装 nvidia-container-toolkit
  • 重启 Docker:sudo systemctl restart docker
  • 验证 GPU 可见:docker run --rm --gpus all nvidia/cuda:11.8-runtime-ubuntu22.04 nvidia-smi

选择并拉取合适的 TensorFlow 官方镜像

TensorFlow 官方镜像托管在 Docker Hub(tensorflow/tensorflow),按用途分几类:

  • CPU 开发:tensorflow/tensorflow:2.13.0-jupyter(含 JupyterLab,适合快速实验)
  • GPU 训练:tensorflow/tensorflow:2.13.0-gpu(含 CUDA 11.8 + cuDNN 8.6,适配主流显卡)
  • 轻量服务:tensorflow/tensorflow:2.13.0-slim(无 Jupyter,体积小,适合部署)

拉取命令示例:docker pull tensorflow/tensorflow:2.13.0-gpu。镜像约 4–5 GB,首次拉取需一定时间。

Docker Hardening
Docker Hardening

专业的Docker安全配置生成器,符合CIS Benchmark v1.8.0标准。

下载

启动容器并挂载必要资源

启动时注意三点:端口映射、目录挂载、GPU 分配。常用命令结构如下:

  • 启动带 Jupyter 的开发环境:docker run -it --gpus all -p 8888:8888 -v $(pwd)/notebooks:/tf/notebooks tensorflow/tensorflow:2.13.0-jupyter
  • 启动纯训练环境(后台运行):docker run -d --gpus all -v /data:/workspace -w /workspace tensorflow/tensorflow:2.13.0-gpu python train.py

说明:

  • --gpus all 启用全部 GPU;也可指定单卡,如 --gpus device=0
  • -v 将本地代码/数据目录挂载进容器,确保训练结果不丢失
  • -p 8888:8888 映射 Jupyter 端口,浏览器访问 http://localhost:8888 即可使用

验证 TensorFlow 是否正常工作

进入容器后,执行简单检查即可确认环境可用:

  • 运行 python -c "import tensorflow as tf; print(tf.__version__, tf.config.list_physical_devices('GPU'))"
  • 输出应显示版本号(如 2.13.0)和 GPU 设备列表(非空表示 GPU 已识别)
  • 若只看到 CPU 设备,请回查 NVIDIA Container Toolkit 是否安装正确、nvidia-smi 在宿主机是否可用

TensorBoard 也已内置,只需在代码中写入 log_dir,并将容器 6006 端口映射到宿主机,即可通过 http://localhost:6006 实时查看训练曲线。

相关文章

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

相关标签:

docker

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
k8s和docker区别
k8s和docker区别

k8s和docker区别有抽象层次不同、管理范围不同、功能不同、应用程序生命周期管理不同、缩放能力不同、高可用性等等区别。本专题为大家提供k8s和docker区别相关的各种文章、以及下载和课程。

2023.07.24

805

4

docker进入容器的方法有哪些
docker进入容器的方法有哪些

docker进入容器的方法:1. Docker exec;2. Docker attach;3. Docker run --interactive --tty;4. Docker ps -a;5. 使用 Docker Compose。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

2024.04.08

6599

6

docker容器无法访问外部网络怎么办
docker容器无法访问外部网络怎么办

docker 容器无法访问外部网络的原因和解决方法:配置 nat 端口映射以将容器端口映射到主机端口。根据主机兼容性选择正确的网络驱动(如 host 或 overlay)。允许容器端口通过主机的防火墙。配置容器的正确 dns 服务器。选择正确的容器网络模式。排除主机网络问题,如防火墙或连接问题。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

2024.04.08

7717

6

docker镜像有什么用
docker镜像有什么用

docker 镜像是预构建的软件组件,用途广泛,包括:应用程序部署:简化部署,提高移植性。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

2024.04.08

3073

7

Docker容器化部署与DevOps实践
Docker容器化部署与DevOps实践

本专题面向后端与运维开发者,系统讲解 Docker 容器化技术在实际项目中的应用。内容涵盖 Docker 镜像构建、容器运行机制、Docker Compose 多服务编排,以及在 DevOps 流程中的持续集成与持续部署实践。通过真实场景演示,帮助开发者实现应用的快速部署、环境一致性与运维自动化。

2026.02.11

339

16

Docker 容器部署
Docker 容器部署

本专题整合了Docker容器部署相关内容,阅读专题下面的文章了解更多详细操作教程。

2026.03.31

491

14

Java容器化部署与Docker实践教程合集
Java容器化部署与Docker实践教程合集

聚焦 Java 应用的容器化与云原生部署,讲解 Dockerfile 编写规范与 Java 应用镜像构建、多阶段构建(Multi-stage Build)减小镜像体积、Jib / Buildpacks 免 Dockerfile 镜像构建方案、JVM 容器感知参数(-XX:MaxRAMPercentage)配置、Docker Compose 编排多服务(应用 + MySQL + Redis)、容器健康检查与资源限制、Kubernetes

2026.05.11

460

26

Go Docker与容器化部署教程合集
Go Docker与容器化部署教程合集

聚焦 Go 应用的容器化部署优势与实践,讲解 Go 静态编译特性(CGO_ENABLED=0)与 scratch / distroless 极小基础镜像构建、多阶段 Dockerfile 编写规范、交叉编译生成目标平台二进制、镜像安全扫描(Trivy)与漏洞修复、Docker Compose 本地编排开发环境、Kubernetes Deployment / Service / ConfigMap 部署 Go 服务、健康检查(Livene

2026.05.15

345

24

Hyperf 项目部署运维与性能调优手册
Hyperf 项目部署运维与性能调优手册

聚焦 Hyperf 在生产环境的落地,包含 Docker 高效打包、Swoole 配置优化、常见的内存溢出(OOM)问题排查方法以及热更新方案。

2026.05.19

545

15

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Docker Desktop故障排查手册
Docker Desktop故障排查手册

共0课时 | 0人学习

Docker镜像构建手册
Docker镜像构建手册

共0课时 | 0人学习