Setup EC2 instance for stable-diffusion-webui

setup_stable-diffusion-webui.sh

# install NVIDIA Drivers
sudo yum update -y
sudo yum install -y kernel-devel

# for amazon linux 2
# sudo update-alternatives --install /usr/bin/cc cc /usr/bin/gcc10-cc 30

# for amazon linux 2023
sudo update-alternatives --install /usr/bin/cc cc /usr/bin/gcc 30

# see: https://docs.nvidia.com/datacenter/tesla/index.html
BASE_URL=https://us.download.nvidia.com/tesla
DRIVER_VERSION=525.105.17
ARCH=$( /bin/arch )
curl -fSsl6 -O $BASE_URL/$DRIVER_VERSION/NVIDIA-Linux-$ARCH-$DRIVER_VERSION.run
sudo sh NVIDIA-Linux-$ARCH-$DRIVER_VERSION.run -a -q -ui=none


# install Docker & nvidia-container-toolkit 
sudo yum install -y docker
sudo usermod -a -G docker ec2-user

distribution=$(. /etc/os-release;echo $ID$VERSION_ID) \
   && curl -sL6 https://nvidia.github.io/libnvidia-container/$distribution/libnvidia-container.repo | sudo tee /etc/yum.repos.d/nvidia-container-toolkit.repo

sudo yum clean expire-cache
sudo yum install -y nvidia-container-toolkit
sudo nvidia-ctk runtime configure --runtime=docker
sudo systemctl restart docker

# install docker-compose
DOCKER_CONFIG=${DOCKER_CONFIG:-$HOME/.docker}
ARCH=$( /bin/arch )
mkdir -p $DOCKER_CONFIG/cli-plugins
curl -SL https://github.com/docker/compose/releases/download/v2.4.1/docker-compose-linux-$ARCH -o $DOCKER_CONFIG/cli-plugins/docker-compose
chmod +x $DOCKER_CONFIG/cli-plugins/docker-compose

### REBOOT HERE ###

# mount ephemeral disk to /workspace
sudo mkfs -t xfs /dev/nvme1n1
sudo mkdir /workspace
sudo mount /dev/nvme1n1 /workspace
sudo chmod o+w /workspace

## move docker images to /workspace
sudo systemctl stop docker.service
sudo mv /var/lib/docker /workspace
sudo ln -s /workspace/docker /var/lib/docker
sudo systemctl start docker.service

# install stable-diffusion-webui-docker
sudo yum install -y git
cd /workspace
git clone https://github.com/AbdBarho/stable-diffusion-webui-docker.git

cd stable-diffusion-webui-docker
docker compose --profile download up --build
docker compose --profile auto up --build