From fafa3e14d6caff36f1cf1a7778cc5a09065f4e0d Mon Sep 17 00:00:00 2001
From: yiyuanzzz <yiyzhong@amazon.com>
Date: Thu, 5 Dec 2024 13:26:36 -0800
Subject: [PATCH] pin the version of nvidia-container-toolkit

---
 scripts/enable-ecs-agent-gpu-support.sh | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/scripts/enable-ecs-agent-gpu-support.sh b/scripts/enable-ecs-agent-gpu-support.sh
index 269bcf6..9677eb8 100644
--- a/scripts/enable-ecs-agent-gpu-support.sh
+++ b/scripts/enable-ecs-agent-gpu-support.sh
@@ -140,10 +140,10 @@ else
         xorg-x11-server-Xorg \
         docker-runtime-nvidia \
         oci-add-hooks \
-        libnvidia-container1 \
-        libnvidia-container-tools \
-        nvidia-container-toolkit-base \
-        nvidia-container-toolkit
+        libnvidia-container1-1.16.2 \
+        libnvidia-container-tools-1.16.2 \
+        nvidia-container-toolkit-base-1.16.2 \
+        nvidia-container-toolkit-1.16.2
 
     sudo yum install -y cuda-drivers \
         cuda