Add welcome message

Author: shub-kris

infrastructure/ami/hcl2-files/build.pkr.hcl

@@ -16,4 +16,14 @@ build {
   provisioner "shell" {
     inline = ["echo 'source /opt/aws_neuron_venv_pytorch/bin/activate' >> /home/ubuntu/.bashrc"]
   }
+  provisioner "file" {
+    source = "scripts/welcome-msg.sh"
+    destination = "/tmp/99-custom-message"
+  }
+  provisioner "shell" {
+    inline = [
+    "sudo mv /tmp/99-custom-message /etc/update-motd.d/",
+    "sudo chmod +x /etc/update-motd.d/99-custom-message",
+  ]
+  }
 }

infrastructure/ami/scripts/welcome-msg.sh

@@ -0,0 +1,11 @@
+#!/bin/bash
+printf "=============================================================================\n"
+printf "       __|  __|_  )\n"
+printf "       _|  (     /   HuggingFace Deep Learning Neuron AMI (Ubuntu 20.04)\n"
+printf "      ___|\___|___|\n"
+printf "=============================================================================\n"
+printf "Welcome to the HuggingFace Deep Learning Neuron AMI (Ubuntu 20.04)\n"
+printf "* Examples:     /home/ubuntu/huggingface-neuron-samples \n"
+printf "* Notebooks:    /home/ubuntu/huggingface-neuron-notebooks \n"
+printf "* Documentation: https://huggingface.co/docs/optimum-neuron/ \n"
+printf "=============================================================================\n"