Docker images are open now

pegasus/README.md

@@ -24,9 +24,9 @@ Otherwise, you can set up Rust [here](https://www.rust-lang.org/tools/install),
 Every node must have two things:
 
 1. This repository cloned under `~/workspace/leaderboard`.
-  - If you want a different path, search and replace in `setup-nodes.yaml`.
+  - If you want a different path, search and replace in `spawn-containers.yaml`.
 2. Model weights under `/data/leaderboard/weights`.
-  - If you want a different path, search and replace in `setup-nodes.yaml` and `benchmark.yaml`.
+  - If you want a different path, search and replace in `setupspawn-containers.yaml` and `benchmark.yaml`.
 
 ### Specify node names for Pegasus
 
@@ -37,11 +37,11 @@ Modify `hosts.yaml` with nodes. See the file for an example.
 
 ### Set up Docker containers on your nodes with Pegasus
 
-This builds our Docker image and spawns one container per GPU (named `leaderboard%d`), for every node.
+This spawns one container per GPU (named `leaderboard%d`), for every node.
 
 ```console
 $ cd pegasus
-$ cp setup-nodes.yaml queue.yaml
+$ cp spawn-containers.yaml queue.yaml
 $ pegasus b
 ```
 

pegasus/{setup-nodes.yaml → spawn-containers.yaml}

@@ -3,5 +3,4 @@
 # {{ gpu }} is defined in `hosts.yaml`, and will be filled in when Pegasus
 # determines the specific node and gpu the generated job command will run on.
 # We check {{ gpu }} = 0 to ensure that the image is only built once on each node.
-- if [ {{ gpu }} = 0 ]; then cd workspace/leaderboard && docker build -t ml-energy:latest .; fi
-- docker run -dit --name leaderboard{{ gpu }} --gpus '"device={{ gpu }}"' -v /data/leaderboard:/data/leaderboard -v $HOME/workspace/leaderboard:/workspace/leaderboard ml-energy:latest bash
+- docker run -dit --name leaderboard{{ gpu }} --gpus '"device={{ gpu }}"' -v /data/leaderboard:/data/leaderboard -v $HOME/workspace/leaderboard:/workspace/leaderboard mlenergy/leaderboard:latest bash