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