Skip to content

Enable GPU runtime support for AWS EC2 instances#275

Merged
yuhuyoyo merged 1 commit intomasterfrom
yuhuyoyo/aws-gpu-support
Dec 1, 2025
Merged

Enable GPU runtime support for AWS EC2 instances#275
yuhuyoyo merged 1 commit intomasterfrom
yuhuyoyo/aws-gpu-support

Conversation

@yuhuyoyo
Copy link
Contributor

@yuhuyoyo yuhuyoyo commented Dec 1, 2025

Summary

  • Remove cloud-specific restriction for GPU runtime configuration
  • Enable NVIDIA GPU support for AWS EC2 instances (g4dn.xlarge, p3.2xlarge)
  • Apply GPU runtime configuration whenever accelerator is set to "nvidia" regardless of cloud platform

Context

Previously, the parse-devcontainer.sh script only applied GPU runtime configuration when both:

  1. Accelerator was set to "nvidia"
  2. Cloud platform was "gcp"

This prevented AWS EC2 GPU instances from getting the proper runtime configuration in their docker-compose.yaml files.

Changes

Modified startupscript/butane/005-parse-devcontainer.sh to check only for accelerator == "nvidia" without restricting to GCP.

Test plan

  • Create AWS EC2 instance with g4dn.xlarge or p3.2xlarge instance type
  • Verify that docker-compose.yaml includes GPU runtime configuration
  • Verify that containers can access NVIDIA GPU resources

🤖 Generated with Claude Code

Previously, the GPU runtime configuration was only applied to GCP instances.
This change enables GPU support for AWS EC2 instances (g4dn.xlarge, p3.2xlarge)
by removing the cloud-specific restriction and applying the nvidia runtime
configuration whenever the accelerator parameter is set to "nvidia".

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude <noreply@anthropic.com>
Copy link
Contributor

@PeterSu92 PeterSu92 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@yuhuyoyo yuhuyoyo merged commit 1a4ed51 into master Dec 1, 2025
23 of 24 checks passed
@yuhuyoyo yuhuyoyo deleted the yuhuyoyo/aws-gpu-support branch December 1, 2025 18:41
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants