on_startup.sh

#!/bin/bash
# Write some commands here that will run on root user before startup.
# For example, to clone transformers and install it in dev mode:
# git clone https://github.com/huggingface/transformers.git
# cd transformers && pip install -e ".[dev]"
git config --global credential.helper store

# Adding more processes/threads
echo '* soft nproc 1024' | sudo tee -a /etc/security/limits.conf
echo '* hard nproc 2048' | sudo tee -a /etc/security/limits.conf

echo 'session required pam_limits.so' | sudo tee -a /etc/pam.d/common-session
echo 'session required pam_limits.so' | sudo tee -a /etc/pam.d/common-session-noninteractive