From f2192695dbfb19d1fa78c1302fb3cca022dbaee0 Mon Sep 17 00:00:00 2001 From: Saisree Valluri Date: Tue, 26 Sep 2023 18:03:05 +0000 Subject: [PATCH] Enable Persistence mode in userdataALTemplate --- scripts/enable-ecs-agent-gpu-support.sh | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/scripts/enable-ecs-agent-gpu-support.sh b/scripts/enable-ecs-agent-gpu-support.sh index 6b1ed18..df274c8 100644 --- a/scripts/enable-ecs-agent-gpu-support.sh +++ b/scripts/enable-ecs-agent-gpu-support.sh @@ -43,9 +43,12 @@ sudo yum install -y kernel-devel-$(uname -r) \ sudo yum install -y cuda-drivers \ cuda -# The Fabric Manager service needs be started and enabled on EC2 P4d instances +# The Fabric Manager service needs to be started and enabled on EC2 P4d instances # in order to configure NVLinks and NVSwitches sudo systemctl enable nvidia-fabricmanager +# NVIDIA Persistence Daemon needs to be started and enabled on P5 instances +# to maintain persistent software state in the NVIDIA driver. +sudo systemctl enable nvidia-persistenced mkdir -p /tmp/ecs echo 'ECS_ENABLE_GPU_SUPPORT=true' >>/tmp/ecs/ecs.config sudo mv /tmp/ecs/ecs.config /var/lib/ecs/ecs.config