chore(ami): use AWS Neuron SDK 2.26 AMI as base

Author: dacorvo

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

@@ -16,7 +16,7 @@ build {
   provisioner "shell" {
     inline = [
       "echo 'export HF_HUB_ENABLE_HF_TRANSFER=1' | sudo tee -a /home/ubuntu/.bashrc",
-      "echo 'source /opt/aws_neuronx_venv_pytorch_2_7/bin/activate' | sudo tee -a /home/ubuntu/.bashrc"
+      "echo 'source /opt/aws_neuronx_venv_pytorch_2_8/bin/activate' | sudo tee -a /home/ubuntu/.bashrc"
     ]
   }
   provisioner "file" {

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

@@ -10,7 +10,7 @@ variable "instance_type" {
 }
 
 variable "source_ami" {
-  default     = "ami-0ffd183ece0ca0475"
+  default     = "ami-0ec4ab14b1c5a10f2"
   description = "Base Image"
   type        = string
   /*

infrastructure/ami/scripts/install-huggingface-libraries.sh

@@ -1,7 +1,7 @@
 #!/bin/bash
 
 # Activate the neuron virtual environment
-source /opt/aws_neuronx_venv_pytorch_2_7/bin/activate
+source /opt/aws_neuronx_venv_pytorch_2_8/bin/activate
 
 echo "Step: install-hugging-face-libraries"
 

infrastructure/ami/scripts/validate-neuron.sh

@@ -3,7 +3,7 @@ echo "Step: validate-neuron-devices"
 neuron-ls
 
 # Activate the neuron virtual environment
-source /opt/aws_neuronx_venv_pytorch_2_7/bin/activate
+source /opt/aws_neuronx_venv_pytorch_2_8/bin/activate
 
 python -c 'import torch'
 python -c 'import torch_neuronx'