ar-mine's picture
Phase 2.1A Robometer-Low training kit v2
e41bef1 verified
|
Raw History Blame Contribute Delete
1.32 kB

Phase 2.1A uv environment

This directory is the deployable Python environment definition for Robometer-Low-LoRA and Robometer-Low. Do not copy a local .venv between machines.

Host requirements

  • Linux x86_64
  • NVIDIA GPU with BF16 support
  • NVIDIA driver compatible with CUDA 12.8
  • uv 0.11.3 or a compatible newer release
  • A CUDA development toolkit is needed only if no compatible prebuilt FlashAttention wheel can be selected and a source build is required.

Rebuild

From the experiment root:

uv python install 3.10.12
uv sync --project environment/phase21a_robometer --frozen
uv run --project environment/phase21a_robometer python environment/phase21a_robometer/doctor.py

The second uv sync --frozen --dry-run must report no changes. Run the formal training smoke test before starting a seed.

The formal trainer sets PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True before initializing CUDA. Keep this setting for the validated micro-batch of 32 pairs; removing it can cause allocator-fragmentation OOM near the RTX 5090 capacity limit even when several GiB are reserved but unused.

The base model, public-proxy dataset, and run output are external artifacts and must not be placed inside this environment directory. Pin their Hugging Face revisions in the run manifest.