Add in AWS Credentials variable for NVIDIA install script (#2037)

Co-authored-by: Wyatt Hicken <[email protected]>
awslabs · Nov 4, 2024 · 974fdf2 · 974fdf2
1 parent 634b772
commit 974fdf2
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/templates/al2023/template.json b/templates/al2023/template.json
@@ -237,6 +237,9 @@
       "environment_vars": [
         "AWS_REGION={{user `aws_region`}}",
         "ENABLE_ACCELERATOR={{user `enable_accelerator`}}",
+        "AWS_ACCESS_KEY_ID={{user `aws_access_key_id`}}",
+        "AWS_SECRET_ACCESS_KEY={{user `aws_secret_access_key`}}",
+        "AWS_SESSION_TOKEN={{user `aws_session_token`}}",
         "BINARY_BUCKET_NAME={{user `binary_bucket_name`}}",
         "BINARY_BUCKET_REGION={{user `binary_bucket_region`}}",
         "NVIDIA_DRIVER_MAJOR_VERSION={{user `nvidia_driver_major_version`}}",