Skip to content
Snippets Groups Projects

Compare revisions

Changes are shown as if the source revision was being merged into the target revision. Learn more about comparing revisions.

Source

Select target project
No results found
Select Git revision
Loading items

Target

Select target project
  • lsds/teaching/continuum-computing/distributed-for-fl
1 result
Select Git revision
Loading items
Show changes
Commits on Source (3)
File added
......@@ -2,7 +2,7 @@ version: '3.8'
services:
node0:
image: local-comm
image: image-dfl-bad
container_name: node0
environment:
- RANK=0
......@@ -15,7 +15,7 @@ services:
node1:
image: local-comm
image: image-dfl-bad
container_name: node1
environment:
- RANK=1
......@@ -28,7 +28,7 @@ services:
node2:
image: local-comm
image: image-dfl-bad
container_name: node2
environment:
- RANK=2
......@@ -41,7 +41,7 @@ services:
node3:
image: local-comm
image: image-dfl-bad
container_name: node3
environment:
- RANK=3
......
import sys
import argparse
image = 'local-comm'
#image = 'image-dfl-bad'
def generate_docker_compose(num_nodes):
def generate_docker_compose(image_name, num_nodes):
docker_compose_template = """\
version: '3.8'
services:
......@@ -17,7 +18,7 @@ services:
- BASE_PORT=8000
- IP_ADDRESS=node{i}
volumes:
- ./neighbors/neighbors{file_suffix}.txt:/app/neighbors.txt
- ./neighbors/neighbors{i}.txt:/app/neighbors.txt
networks:
- my_network
......@@ -31,9 +32,8 @@ networks:
docker_compose_content = docker_compose_template
for i in range(0, num_nodes):
file_suffix = i
docker_compose_content += service_template.format(i=i, file_suffix=file_suffix,img=image)
for i in range(0, int(num_nodes)):
docker_compose_content += service_template.format(i=i,img=image_name)
docker_compose_content += network_section
......@@ -41,9 +41,17 @@ networks:
file.write(docker_compose_content)
if __name__ == "__main__":
if len(sys.argv) != 2:
print("Usage: python generate_docker_compose.py <num_nodes>")
sys.exit(1)
num_nodes = int(sys.argv[1])
generate_docker_compose(num_nodes)
parser = argparse.ArgumentParser(description="Upload PDF files to an S3 bucket")
parser.add_argument("--image_name", help="The name of the image")
parser.add_argument("--nb_nodes", help="The number of nodes")
args = parser.parse_args()
# image_name= args.image_name
# if len(sys.argv) != 2:
# print("Usage: python generate_docker_compose.py <num_nodes>")
# sys.exit(1)
# num_nodes = int(sys.argv[1])
generate_docker_compose(args.image_name, args.nb_nodes)