update to use neuron v2 (#70)

aws · Sep 6, 2022 · 26a36b9 · 26a36b9
1 parent e48ffb9
commit 26a36b9
Showing 1 changed file with 2 additions and 12 deletions.
diff --git a/scripts/enable-ecs-agent-inferentia-support.sh b/scripts/enable-ecs-agent-inferentia-support.sh
@@ -27,12 +27,8 @@ sudo mv /tmp/neuron.repo /etc/yum.repos.d/neuron.repo
 sudo yum install kernel-devel-$(uname -r) kernel-headers-$(uname -r) -y
 
 # Install Neuron Driver
-sudo yum install -y aws-neuron-dkms
-# Install Neuron Tools
-sudo yum install -y aws-neuron-tools
-sudo yum install -y aws-neuron-runtime-base
-
-sudo yum install -y pciutils oci-add-hooks
+sudo yum install -y aws-neuronx-dkms-2.*
+sudo yum install -y aws-neuronx-oci-hook-2.*
 
 # disable neuron package upgrades by deleting the yum repo
 sudo rm /etc/yum.repos.d/neuron.repo
@@ -48,9 +44,3 @@ if [ ! -f $NEURON_RUNTIME ]; then
     sudo cp /opt/aws/neuron/bin/oci_neuron_hook_wrapper.sh $NEURON_RUNTIME
     sudo chmod +x $NEURON_RUNTIME
 fi
-
-# Copy neuron-discovery which is a prereq for neuron to function
-if [ ! -f /lib/systemd/system/neuron-discovery.service ]; then
-    sudo cp /opt/aws/neuron/config/neuron-discovery.service /lib/systemd/system/
-    sudo systemctl enable neuron-discovery
-fi