pin the version of nvidia-container-toolkit

aws · Dec 6, 2024 · dfe09af · dfe09af
1 parent 4b0cd07
commit dfe09af
Showing 1 changed file with 4 additions and 4 deletions.
diff --git 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