echen9898/curiosity_baselines

echen9898/curiosity_baselines 镜像说明

镜像概述

这是一个用于运行内在动机强化学习算法的Docker镜像，基于nvidia/cuda:10.2-base-ubuntu18.04构建，支持GPU加速，集成多种强化学习环境和必要依赖，适用于强化学习领域的研究与开发。

核心功能

• 多环境支持：涵盖Mujoco、Metaworld、Mazeworld、pycolab等主流强化学习环境；
• GPU加速：基于CUDA 10.2，可利用NVIDIA GPU实现高效采样与训练；
• 可视化工具：集成matplotlib等绘图库，便于实验结果分析；
• 并行计算：包含mpi4py库，支持分布式训练场景；
• 版本迭代：不同标签版本（如0.1.7_gpu）对应环境与依赖的更新优化。

使用场景

• 内在动机强化学习算法的研究与实现；
• 多RL环境下的算法性能测试与评估；
• 强化学习模型的快速部署与实验验证。

配置说明

• 硬件要求：主机需配备NVIDIA GPU，并安装与CUDA 10.2兼容的驱动；
• 版本选择：根据需求选择对应标签（如0.1.7_gpu支持*** Control环境）；
• 依赖预装：镜像已包含gym、mpi4py等Python包及Linux系统依赖。

部署示例

Docker Run命令

bash
docker run -it --gpus all echen9898/curiosity_baselines:0.1.7_gpu bash

该命令启动镜像并进入交互模式，利用所有可用GPU资源。

版本历史

• 0.1.9: 调整本地包导入路径；
• 0.1.8: 添加matplotlib及绘图依赖；
• 0.1.7_gpu: 新增*** Control环境支持；
• 0.1.7: 更新Mazeworld环境；
• 0.1.6_gpu: 新增Metaworld环境支持；
• 0.1.6: 添加pycolab和Mazeworld环境；
• 0.1.5_gpu: 新增公共Mujoco许可证及稀疏Mujoco环境支持；
• 0.1.5: 添加mpi4py；
• 0.1.4_gpu: 新增新环境所需Python包；
• 0.1.4: 添加基础Python包；
• 0.1.3_gpu: 调整本地包导入路径；
• 0.1.3: 清理0.1.2版本内容；
• 0.1.2_gpu: 添加matplotlib及绘图依赖；
• 0.1.2: 移除vizdoom和gym-retro，新增gym-super-mario-bros及部分Linux包；
• 0.1.1: 新增gym-retro和vizdoom环境支持；
• 0.1.0_gpu: 基于nvidia/cuda:10.2-base-ubuntu18.04构建，支持GPU采样；
• 0.1.0: 运行rlpyt代码库及基础环境。