rocker/ml-verse

Rocker CUDA 镜像文档

镜像概述

本镜像由 https://github.com/rocker-org/rocker-versioned2 项目构建，是Rocker项目提供的支持CUDA的版本化镜像。该镜像将R语言运行环境与CUDA工具包集成，旨在为数据科学和科学计算领域提供具备GPU加速能力的可靠运行环境。

核心功能与特性

• 完整R环境：包含R语言核心及常用科学计算库，支持标准R脚本执行和包管理
• CUDA支持：集成NVIDIA CUDA工具包，可利用GPU加速计算密集型任务
• 版本化管理：提供稳定的版本控制，确保开发和生产环境一致性
• 兼容性：兼容主流Linux系统和NVIDIA GPU驱动，支持Docker GPU运行时
• 轻量级设计：基于官方基础镜像优化，平衡功能完整性和资源占用

使用场景与适用范围

• 需要GPU加速的R语言数据分析任务
• 大规模统计建模和数值计算
• 基于R的机器学习和深度学习项目（如使用keras、tensorflow等R包）
• 学术研究中的高性能计算场景
• 数据科学教学与实验环境搭建

使用方法与配置说明

前提条件

• 已安装Docker Engine（19.03+版本）
• 已配置NVIDIA Docker运行时（nvidia-docker2）
• 系统已安装兼容的NVIDIA GPU驱动

镜像拉取

bash
# 拉取最新版本
docker pull rocker/cuda:latest

# 拉取特定版本（例如CUDA 11.7, R 4.2）
docker pull rocker/cuda:4.2.3-cuda11.7

基本运行示例

以交互模式启动R终端，启用GPU支持：

bash
docker run -it --gpus all rocker/cuda:latest R

运行Jupyter Notebook（如镜像包含）

bash
docker run -it --gpus all -p 8888:8888 rocker/cuda:latest jupyter notebook --ip=0.0.0.0 --allow-root

挂载本地目录

持久化工作数据和脚本：

bash
docker run -it --gpus all -v /本地工作目录:/home/rstudio/work rocker/cuda:latest

环境变量配置

常用环境变量：

• R_VERSION：指定R语言版本（如"4.3.0"）
• CUDA_VERSION：指定CUDA版本（如"11.8"）
• USER：运行用户（默认为"rstudio"或"root"）
• PASSWORD：设置rstudio用户密码（如适用）

示例：

bash
docker run -it --gpus all -e PASSWORD=yourpassword rocker/cuda:latest

更多信息

• 官方文档：Rocker CUDA镜像说明
• 源代码仓库：https://github.com/rocker-org/rocker-versioned2
• 版本标签说明：镜像标签格式通常为[R版本]-cuda[CUDA版本]，具体版本可在Docker Hub查看