datamachines/cudnn_tensorflow_opencv

cudnn_tensorflow_opencv Docker镜像文档

镜像概述

该Docker镜像基于Nvidia GPU环境构建，集成了CUDA、CuDNN、TensorFlow、OpenCV等核心组件，旨在为深度学习和计算机视觉任务提供开箱即用的GPU加速环境。镜像构建自GitHub仓库https://github.com/datamachines/cuda_tensorflow_opencv%EF%BC%8C%E7%9B%B8%E5%85%B3Dockerfile%E5%8F%8A%E6%9B%B4%E5%A4%9A%E9%95%9C%E5%83%8F%E5%8F%AF%E5%9C%A8Docker Hubhttps://hub.docker.com/r/datamachines/tensorflow_opencv获取。

标签命名规则

镜像标签遵循cuda_tensorflow_opencv顺序命名，格式为：[CUDA版本]_[TensorFlow版本]_[OpenCV版本]-[构建日期]，其中：

• [CUDA版本]：包含CuDNN支持的CUDA版本号
• [TensorFlow版本]：集成的TensorFlow版本
• [OpenCV版本]：集成的OpenCV版本
• [构建日期]：Dockerfile的构建日期（YYYYMMDD格式）

示例：cudnn_tensorflow_opencv:10.2_1.15_4.1.2-20191210表示基于2019年12月10日Dockerfile构建，包含CUDA 10.2（CuDNN7）、TensorFlow 1.15和OpenCV 4.1.2。

核心功能与特性

• 完整的GPU加速栈：集成CUDA、CuDNN，支持Nvidia GPU硬件加速
• 主流深度学习框架：包含TensorFlow，支持模型训练与推理
• 计算机视觉工具：集成OpenCV，支持图像处理与计算机视觉任务
• 多媒体处理：包含FFmpeg，支持视频编解码
• 多版本支持：提供不同CUDA、TensorFlow、OpenCV版本组合，适配不同应用需求
• 稳定基础环境：基于Ubuntu 20.04构建，确保系统稳定性

使用场景

• 深度学习模型开发与训练（尤其是计算机视觉相关任务）
• 计算机视觉应用原型开发与测试
• GPU加速的图像处理与视频分析
• 依赖TensorFlow和OpenCV的生产环境部署

使用方法

前提条件

• 已安装Docker Engine
• 已安装Nvidia Docker运行时（nvidia-docker2），确保GPU支持

拉取镜像

bash
docker pull datamachines/tensorflow_opencv:[标签]

例如，拉取20220815版本中CUDA 11.3.1、TensorFlow 2.9.1、OpenCV 4.6.0的镜像：

bash
docker pull datamachines/tensorflow_opencv:11.3.1_2.9.1_4.6.0-20220815

运行容器

基本运行命令（启用GPU支持）：

bash
docker run --gpus all -it --rm datamachines/tensorflow_opencv:11.3.1_2.9.1_4.6.0-20220815 /bin/bash

挂载本地目录（将主机/path/to/data目录挂载到容器/data）：

bash
docker run --gpus all -it --rm -v /path/to/data:/data datamachines/tensorflow_opencv:11.3.1_2.9.1_4.6.0-20220815 /bin/bash

Docker Compose配置示例

yaml
version: '3'
services:
  cv-tf-gpu:
    image: datamachines/tensorflow_opencv:11.3.1_2.9.1_4.6.0-20220815
    runtime: nvidia
    volumes:
      - ./data:/app/data
    command: python /app/train.py

20220815版本可用标签详情

更多信息

• GitHub仓库README：https://github.com/datamachines/cuda_tensorflow_opencv/blob/master/README.md
• 完整标签列表：https://github.com/datamachines/cuda_tensorflow_opencv/blob/master/Builds-DockerHub.md#cudnn_tensorflow_opencv