diff --git a/manifest.yaml b/manifest.yaml new file mode 100644 index 00000000..7f966937 --- /dev/null +++ b/manifest.yaml @@ -0,0 +1,34 @@ +# yaml-language-server: $schema=https://instadeep.aichor.ai/schema/latest/manifest.schema.json + +kind: AIchorManifest +apiVersion: 0.2.0 + +builder: + image: og-marl + dockerfile: ./Dockerfile + context: . + buildArgs: + USE_CUDA: "true" + +spec: + operator: tf + image: og-marl + command: wandb agent off-the-grid-marl-team/og-marl-baselines/m5wro67c + # command: wandb agent off-the-grid-marl-team/og-marl-baselines/i87akim6 + + types: + Worker: + count: 60 + + resources: + # 10-15 cpus for single process + cpus: 8 + # mandatory (must be >= 2): RAM ratio wanted (in GB) + # memory = cpus * ramRatio + ramRatio: 3 + accelerators: + gpu: + count: 0 + # mandatory (if gpus amount is > 0): gpu type wanted + type: gpu + product: Quadro-RTX-4000 # Quadro-RTX-4000 | Tesla-V100-SXM3-32GB