跳转到内容

环境配置说明

Terminal window
git clone --recurse-submodules https://github.com/OpenEnvision/WorldAtlas.git WorldAtlas
cd WorldAtlas

若已克隆但未拉子模块:

Terminal window
git submodule update --init --recursive

完整子模块 ↔ 模型配置对照见 thirdparty/README.md。

在运行 evaluation 或模型 adapter 之前,在仓库根目录创建 .env 文件。WorldAtlas 实际读取的变量如下:

Terminal window
# 仅当 ckpt/ 不在 checkout 内时需要
WORLDATLAS_CHECKPOINT_ROOT=/path/to/ckpt
# 建议使用已验证的 CUDA 12.1 栈
CUDA_HOME=/path/to/cuda-12.1

WORLDATLAS_CHECKPOINT_ROOT 是唯一的 checkpoint 根目录。未设置时,所有 checkpoint 类路径都解析到 WorldAtlas/ckpt;设置后必须保持相同的目录布局(huggingface/、torch/、official_metrics/、各模型目录等)。

建议同时导出 HF/Torch 缓存(worldatlas CLI 也会自动应用):

Terminal window
export HF_HOME="${WORLDATLAS_CHECKPOINT_ROOT:-ckpt}/huggingface"
export HUGGINGFACE_HUB_CACHE="${HF_HOME}/hub"
export TRANSFORMERS_CACHE="${HF_HOME}/hub"
export TORCH_HOME="${WORLDATLAS_CHECKPOINT_ROOT:-ckpt}/torch"

创建 .env 后,在每个将要运行 WorldAtlas 的 shell 中导出变量:

Terminal window
set -a
source .env
set +a
export $(grep -v '^#' .env | xargs)

如果要评测 config/models/ 中的 API 模型,把 provider key 放在本地 .secrets 文件或 shell 环境变量中。WorldAtlas 不会提交 API 凭证。