feat: multinode setup

This commit is contained in:
Zach Nussbaum
2023-04-05 02:53:04 +00:00
parent 0a3834d086
commit 399a65e779
3 changed files with 33 additions and 0 deletions

8
create_hostname.sh Normal file
View File

@@ -0,0 +1,8 @@
#!/bin/bash
export WORKER_IP=$1
N_GPUS=8
# create dir if doesn't exist
sudo mkdir -p /job
printf "localhost slots=$N_GPUS\n$WORKER_IP slots=$N_GPUS" | sudo tee /job/hostfile
echo /job/hostfile