ikeyasu/cuda

docker-cuda 镜像文档

镜像概述

docker-cuda是基于NVIDIA官方CUDA镜像修改的Docker镜像，集成了CUDA工具包和cuDNN库，旨在为用户提供便捷的GPU加速计算环境。该镜像在NVIDIA官方镜像基础上进行了优化和调整，确保更好的兼容性和易用性，适用于各类需要GPU加速的应用场景。

核心功能与特性

• 完整CUDA支持：包含完整的CUDA工具包，支持GPU加速计算
• cuDNN集成：内置cuDNN库，优化深度学习框架的GPU加速性能
• 基于官方镜像：在NVIDIA官方CUDA镜像基础上构建，保证核心组件的稳定性
• 多版本支持：提供不同CUDA版本的镜像标签，满足不同应用需求
• 兼容性优化：针对常见GPU加速场景进行配置优化，减少环境配置复杂度

使用场景

• 深度学习模型训练与推理
• 科学计算与数值模拟
• GPU加速的机器学习应用开发
• 需要CUDA环境的应用测试与部署
• 多版本CUDA环境的快速切换与隔离

使用方法

前置条件

• 已安装Docker Engine
• 已安装NVIDIA Docker运行时（nvidia-docker）
• 系统配备支持CUDA的NVIDIA GPU

获取镜像

通过Docker Hub拉取镜像（假设镜像已发布至Docker Hub，具体标签需参考实际仓库）：

bash
docker pull ikeyasu/cuda:[tag]

其中[tag]为具体版本标签，例如：11.7.1-cudnn8-runtime-ubuntu20.04

运行容器

基本运行命令：

bash
docker run --gpus all -it --rm ikeyasu/cuda:[tag]

参数说明：

• --gpus all：允许容器使用所有GPU设备
• -it：交互式终端模式
• --rm：容器退出后自动删除

高级配置

挂载数据卷

将本地目录挂载到容器中，便于数据共享：

bash
docker run --gpus all -it --rm -v /本地目录:/容器目录 ikeyasu/cuda:[tag]

设置环境变量

可通过-e参数设置环境变量，例如：

bash
docker run --gpus all -it --rm -e CUDA_VISIBLE_DEVICES=0 ikeyasu/cuda:[tag]

其中CUDA_VISIBLE_DEVICES=0指定仅使用第0块GPU

验证环境

容器启动后，可通过以下命令验证CUDA环境：

bash
nvcc --version  # 查看CUDA版本
nvidia-smi      # 查看GPU信息

版本标签说明

镜像标签通常包含以下信息：

• CUDA版本（如11.7.1）
• cuDNN版本（如cudnn8）
• 镜像类型（runtime或devel，分别对应运行时和开发环境）
• 基础操作系统（如ubuntu20.04）

例如：11.7.1-cudnn8-runtime-ubuntu20.04表示基于Ubuntu 20.04，包含CUDA 11.7.1和cuDNN 8的运行时环境镜像。

注意事项

• 确保主机已安装匹配的NVIDIA驱动，驱动版本需支持所使用的CUDA版本
• 不同CUDA版本对GPU架构有不同要求，选择时需匹配硬件支持
• 使用--gpus参数时需确保Docker版本≥19.03且已正确配置nvidia-docker运行时