32a241c28f
Some checks failed
nightly / Test against latest dependencies (py3.10) (push) Has been cancelled
nightly / Test against latest dependencies (py3.13) (push) Has been cancelled
tests / tests (3.13, locked) (push) Has been cancelled
tests / tests (3.13, unlocked) (push) Has been cancelled
tests / pyright (3.10) (push) Has been cancelled
tests / lint-format (push) Has been cancelled
tests / tests (3.10, locked) (push) Has been cancelled
tests / tests (3.11, locked) (push) Has been cancelled
tests / tests (3.12, locked) (push) Has been cancelled
tests / pyright (3.11) (push) Has been cancelled
tests / pyright (3.12) (push) Has been cancelled
tests / pyright (3.13) (push) Has been cancelled
tests / ty-check (3.10) (push) Has been cancelled
tests / ty-check (3.11) (push) Has been cancelled
tests / ty-check (3.12) (push) Has been cancelled
tests / ty-check (3.13) (push) Has been cancelled
tests / stubs (push) Has been cancelled
tests / smoke-test (push) Has been cancelled
Docker / check_paths (push) Has been cancelled
docs / build (push) Has been cancelled
Docker / build (push) Has been cancelled
Upstream: https://github.com/michaelgillett/mjlab Upstream-Commit: c19f713c415a699a79d71cd96aa13c3104a05047 Upstream-Branch: main
35 lines
768 B
YAML
35 lines
768 B
YAML
# SkyPilot cluster definition for W&B sweep agents.
|
|
#
|
|
# Provisions a multi-GPU instance and installs dependencies. Does not
|
|
# start any jobs. Use sweep-agent.yaml with sky exec for that.
|
|
#
|
|
# Usage:
|
|
# sky launch scripts/cloud/sweep-cluster.yaml -c mjlab-sweep --gpus A100:8
|
|
|
|
name: mjlab-sweep-cluster
|
|
|
|
resources:
|
|
cloud: lambda
|
|
accelerators: A100:8
|
|
autostop:
|
|
idle_minutes: 5
|
|
down: true
|
|
|
|
workdir: .
|
|
|
|
file_mounts:
|
|
~/.netrc: ~/.netrc
|
|
|
|
envs:
|
|
MUJOCO_GL: egl
|
|
|
|
setup: |
|
|
# Install EGL for MuJoCo headless rendering.
|
|
sudo apt-get update && sudo apt-get install -y libegl-dev
|
|
|
|
# Install uv if not present.
|
|
command -v uv || curl -LsSf https://astral.sh/uv/install.sh | sh
|
|
source "$HOME/.local/bin/env" 2>/dev/null || true
|
|
|
|
uv sync --locked --no-dev
|