Ryan Romanchuk rromanchuk
bilalmughal
/ ec2_graviton_dl_bootstrap.sh
Last active
April 2, 2025 15:29
This script automates the setup of an Amazon EC2 Graviton ARM-based instances for deep learning tasks. It takes care of installing essential utilities, setting up latest Nvidia drivers and CUDA 12.2 toolkit and cuDNN library, and build PyTorch from source. The step-by-step guided can be checked here. https://jumpshare.com/blog/deep-learning-on-a…
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| #!/bin/bash | |
| set -e # Exit on any error | |
| # Check if required arguments are provided | |
| if [ -z "$REGION" ] || [ -z "$SECURITY_GROUPS" ] || [ -z "$KEY_PAIR" ] || [ -z "$SUBNET" ]; then | |
| echo "Error: You must provide REGION, SECURITY_GROUPS, KEY_PAIR, and SUBNET as environment variables." | |
| echo "Example:" | |
| echo " export REGION=us-east-1" | |
| echo " export SECURITY_GROUPS=sg-12345678,sg-87654321" | |
| echo " export KEY_PAIR=my-key-pair" |
OlderNewer