forked from ray-project/ray
-
Notifications
You must be signed in to change notification settings - Fork 0
/
build-multinode-image.py
35 lines (26 loc) · 962 Bytes
/
build-multinode-image.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
import argparse
import os
import shutil
import subprocess
import tempfile
def build_multinode_image(source_image: str, target_image: str):
"""Build docker image from source_image.
This docker image will contain packages needed for the fake multinode
docker cluster to work.
"""
tempdir = tempfile.mkdtemp()
dockerfile = os.path.join(tempdir, "Dockerfile")
with open(dockerfile, "wt") as f:
f.write(f"FROM {source_image}\n")
f.write("RUN sudo apt update\n")
f.write("RUN sudo apt install -y openssh-server\n")
subprocess.check_output(
f"docker build -t {target_image} .", shell=True, cwd=tempdir
)
shutil.rmtree(tempdir)
if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.add_argument("source_image", type=str)
parser.add_argument("target_image", type=str)
args = parser.parse_args()
build_multinode_image(args.source_image, args.target_image)