pywr/hydra-pywr Docker 镜像 - 轩辕镜像

pywr-hydra模型运行环境镜像

概述

本Docker镜像预集成pywr和hydra-pywr软件包，提供一个开箱即用的环境用于运行基于Hydra框架的pywr模型。pywr是一个灵活的Python库，用于水资源系统建模与分析；hydra-pywr则是与Hydra配置框架集成的扩展，支持通过结构化配置文件管理pywr模型参数，简化多场景模拟和参数调优流程。

核心功能与特性

• 预配置依赖：已安装pywr和hydra-pywr核心包，无需手动解决Python环境依赖
• Hydra集成：支持通过Hydra的配置系统管理模型参数、多运行配置和环境隔离
• 轻量级设计：基于Python基础镜像构建，体积小巧，启动快速
• 灵活挂载：支持外部模型文件和数据挂载，便于模型开发与测试

使用场景

• 水资源系统建模与模拟
• 水文模型参数敏感性分析
• 多场景水资源规划评估
• 科研与教育中的水文模型教学演示

使用方法与配置说明

基本运行命令

将本地模型目录挂载至容器，执行hydra模型运行命令：

bash
docker run -v /path/to/local/models:/models [镜像名称] hydra run -m /models/model_config.yaml

参数说明

• -v /path/to/local/models:/models：挂载本地模型目录到容器内/models路径，确保容器可访问模型配置文件（如.yaml）和输入数据
• hydra run -m /models/model_config.yaml：通过hydra-pywr执行指定的模型配置文件（需替换为实际配置文件路径）

自定义模型参数

通过Hydra的命令行参数覆盖功能调整模型参数：

bash
docker run -v /path/to/local/models:/models [镜像名称] hydra run -m /models/model_config.yaml model.reservoir.capacity=1000 model.demand=50

多场景运行

利用Hydra的多运行功能批量执行不同场景：

bash
docker run -v /path/to/local/models:/models [镜像名称] hydra run -m /models/model_config.yaml --multirun scenario=baseline,drought,flood

注意事项

• 确保本地模型目录包含完整的配置文件和依赖数据
• 如需扩展功能（如额外Python库），可基于此镜像构建自定义Dockerfile
• 运行复杂模型时建议分配足够的容器资源（如内存、CPU）以确保性能