openshift-ai-workload.sh

openshift-ai-workload.sh

#!/bin/bash

if [ -z "$1" ]; then
  echo "Instance size not provided. Please pass the instance size as an argument."
  echo "Example: ./openshift-ai-workload.sh m6i.4xlarge"
  echo "Example: ./openshift-ai-workload.sh g4dn.4xlarge"
  echo "Example: ./openshift-ai-workload.sh g4ad.4xlarge"
  exit 1
fi

print_aws_variables() {
    echo "Export the following AWS variables before running this script:"
    echo "export aws_access_key_id=\"YOUR_ACCESS_KEY_ID\""
    echo "export aws_secret_access_key=\"YOUR_SECRET_ACCESS_KEY\""
    echo "export aws_region=\"YOUR_AWS_REGION\""
}

if [  -z ${aws_access_key_id} ]
then
    echo "aws_access_key_id is not set"
    print_aws_variables
    exit 1
fi

if [  -z ${aws_secret_access_key} ]
then
    echo "aws_secret_access_key is not set"
    print_aws_variables
    exit 1
fi
 
if [  -z ${aws_region} ]
then
    echo "aws_region is not set"
    print_aws_variables
    exit 1
fi


if ! yq -v  &> /dev/null
then
    VERSION=v4.34.1
    BINARY=yq_linux_amd64
    sudo wget https://github.com/mikefarah/yq/releases/download/${VERSION}/${BINARY} -O /usr/bin/yq &&\
    sudo chmod +x /usr/bin/yq
fi

# Check if 'oc' is installed
if ! command -v oc &> /dev/null; then
  echo "'oc' (OpenShift command-line tool) is not installed. Installing and configuring..."
  
  # Download the 'configure-openshift-packages.sh' script
  curl -OL https://raw.githubusercontent.com/tosin2013/openshift-4-deployment-notes/master/pre-steps/configure-openshift-packages.sh

  # Make the script executable
  chmod +x configure-openshift-packages.sh 

  # Run the 'configure-openshift-packages.sh' script with the -i flag for installation
  ./configure-openshift-packages.sh -i
else
  echo "'oc' (OpenShift command-line tool) is already installed. Skipping installation."
fi

# Check if '~/.ssh/cluster-key' exists, and if not, generate the SSH key
ssh_key_file="/home/$USER/.ssh/cluster-key"
if ! file_exists "$ssh_key_file"; then
  echo "SSH key '/home/$USER/.ssh/cluster-key' does not exist. Generating..."
  ssh-keygen -t rsa -b 4096 -f /home/$USER/.ssh/cluster-key -N ''
else
  echo "SSH key '/home/$USER/.ssh/cluster-key' already exists. Skipping key generation."
fi

# Check if 'configure-aws-cli.sh' exists in the home directory
script_file="configure-aws-cli.sh"
home_dir="$HOME"

if [ -e "$home_dir/$script_file" ]; then
  echo "The script '$script_file' already exists in the home directory."
  echo "Skipping download and execution."
else
  # Download the 'configure-aws-cli.sh' script
  curl -OL https://raw.githubusercontent.com/tosin2013/openshift-4-deployment-notes/master/aws/$script_file

  # Make the script executable
  chmod +x "$script_file"


  # Run the script
  ./"$script_file" -i $aws_access_key_id $aws_secret_access_key $aws_region || exit 1
fi



openshift-install create install-config --dir=cluster

# Set the default instance size or read from the first script argument
instance_size=${1:-"m6i.4xlarge"}

## NVIDIA GPUs (G4dn) 
# https://instances.vantage.sh/aws/ec2/g4dn.4xlarge

## AMD GPUs (G4ad)
# https://instances.vantage.sh/aws/ec2/g4ad.4xlarge

## Without (GPU)
# m6i.4xlarge

echo "Using AWS Instance Size: $instance_size"
yq -i eval '.compute[0].hyperthreading = "Enabled" |
     .compute[0].name = "worker" |
     .compute[0].platform.aws.rootVolume.iops = 2000 |
     .compute[0].platform.aws.rootVolume.size = 500 |
     .compute[0].platform.aws.rootVolume.type = "io1" |
     .compute[0].platform.aws.type = "'"${instance_size}"'" | 
     .compute[0].replicas = 5' cluster/install-config.yaml
         
openshift-install create cluster --dir $HOME/cluster --log-level debug

# openshift-install destroy cluster --dir=cluster --log-level debug