Make bootstrap GPU helper container runtime aware (#1858)

awslabs · Jun 21, 2024 · 8c6ac67 · 8c6ac67
1 parent 768cbca
commit 8c6ac67
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/templates/al2/runtime/bootstrap.sh b/templates/al2/runtime/bootstrap.sh
@@ -639,7 +639,7 @@ fi
 BOOTSTRAP_GPU_HELPER=/etc/eks/bootstrap-gpu.sh
 if [ -x "${BOOTSTRAP_GPU_HELPER}" ]; then
   log "INFO: starting GPU bootstrap helper..."
-  "${BOOTSTRAP_GPU_HELPER}"
+  $BOOTSTRAP_GPU_HELPER -r "$CONTAINER_RUNTIME"
   log "INFO: completed GPU bootstrap helper!"
 fi