AWS Deep Learning Containers MCP Server
OfficialSupports discovering PyTorch Deep Learning Containers images, provides framework-specific best practices and optimization tips for training and inference.
Supports discovering TensorFlow Deep Learning Containers images, provides framework-specific best practices and optimization tips for training and inference.
Click on "Install Server".
Wait a few minutes for the server to deploy. Once ready, it will show a "Started" state.
In the chat, type
@followed by the MCP server name and your instructions, e.g., "@AWS Deep Learning Containers MCP ServerWhat instance should I use for a 35GB model?"
That's it! The server will respond to your query, and you can continue using it as needed.
Here is a step-by-step guide with screenshots.
AWS Deep Learning Containers MCP Server
A Model Context Protocol (MCP) server for AWS Deep Learning Containers (DLC) that provides tools for discovering, building, deploying, and troubleshooting DLC images.
Features
Dynamic DLC Image Discovery: Automatically fetches latest images from AWS DLC GitHub - always up-to-date
Image Building: Create custom Dockerfiles and build images based on DLC base images
Multi-Platform Deployment: Deploy to SageMaker, EC2, ECS, and EKS
Instance Recommendations: Get GPU instance recommendations based on model size and budget
Upgrade Support: Analyze upgrade paths and generate migration Dockerfiles
Troubleshooting: Diagnose common DLC issues with actionable solutions
Best Practices: Security, cost optimization, and deployment guidance
No AWS Credentials Required: Discovery tools work without AWS credentials
Related MCP server: AWS Documentation MCP Server
Quick Start
Option 1: Run with uv (Recommended)
# Clone the repo
git clone https://github.com/aws-samples/sample-dlc-mcp-server.git
cd sample-dlc-mcp-server
# Run directly with uv
uv run dlc-mcp-serverOption 2: Run with Docker
# Build the image
docker build -t dlc-mcp-server .
# Run the container
docker run -it --rm \
-v ~/.aws:/root/.aws:ro \
dlc-mcp-serverOption 3: Install locally
pip install -e .
dlc-mcp-serverMCP Client Configuration
For Amazon Q CLI
Add to ~/.aws/amazonq/mcp.json:
{
"mcpServers": {
"dlc-mcp-server": {
"command": "uv",
"args": ["--directory", "/path/to/sample-dlc-mcp-server", "run", "dlc-mcp-server"],
"timeout": 120000
}
}
}For Kiro
Add to .kiro/settings/mcp.json:
{
"mcpServers": {
"dlc-mcp-server": {
"command": "uv",
"args": ["--directory", "/path/to/sample-dlc-mcp-server", "run", "dlc-mcp-server"],
"timeout": 120000
}
}
}Using Docker
{
"mcpServers": {
"dlc-mcp-server": {
"command": "docker",
"args": ["run", "-i", "--rm", "-v", "~/.aws:/root/.aws:ro", "dlc-mcp-server"],
"timeout": 120000
}
}
}Available Tools
DLC Discovery
Tool | Description |
| Search DLC images by framework, version, accelerator, platform |
| Get image recommendations based on model type and size |
| List all available frameworks with versions |
| Compare vLLM, SGLang, DJL, NeuronX options |
| Side-by-side image comparison |
| Force refresh image catalog from GitHub |
Image Building
Tool | Description |
| Generate Dockerfile with custom packages |
| Build and optionally push to ECR |
Deployment
Tool | Description |
| Deploy to SageMaker endpoint |
| Launch EC2 instance with DLC |
| Deploy to ECS cluster |
| Deploy to EKS cluster |
| Check endpoint status |
Instance Advisor
Tool | Description |
| GPU instance recommendations by model size |
| List available GPU instances with pricing |
| Estimate training job costs |
Troubleshooting
Tool | Description |
| Analyze error logs with root cause analysis |
| Diagnose common DLC problems |
| Check framework version compatibility |
Best Practices
Tool | Description |
| Security guidelines |
| Cost reduction strategies |
| Platform-specific guidance |
| Framework optimization tips |
Supported Frameworks
Framework | Latest Version | Use Cases |
PyTorch | 2.9.0 | Training, Inference |
TensorFlow | 2.19.0 | Training, Inference |
vLLM | 0.15.1 | LLM Inference |
SGLang | 0.5.8 | LLM Inference |
HuggingFace PyTorch | 2.6.0 | NLP Training/Inference |
AutoGluon | 1.5.0 | AutoML |
DJL | 0.36.0 | Large Model Inference |
PyTorch NeuronX | 2.9.0 | Trainium/Inferentia |
Example Usage
Find vLLM images
Search for vLLM images for SageMaker inferenceDeploy LLM to SageMaker
Deploy Qwen2.5-32B using vLLM on SageMaker with the right instance typeGet instance recommendations
What instance should I use for a 35GB model?Troubleshoot errors
Help me fix this CUDA out of memory error: [paste error]Configuration
Environment variables:
Variable | Description | Default |
| Enable build/deploy operations |
|
| Enable detailed logs access |
|
| Logging level |
|
| Log file path | None |
Development
# Install dev dependencies
pip install -e ".[dev]"
# Run tests
python -m pytest tests/ -v
# Run linting
ruff check .See DEVELOPMENT.md for more details.
License
This library is licensed under the MIT-0 License.
This server cannot be installed
Maintenance
Resources
Unclaimed servers have limited discoverability.
Looking for Admin?
If you are the server author, to access and configure the admin panel.
Latest Blog Posts
- Your AI Chatbot Just Exposed Your CEO's Salary to an InternBy Om-Shree-0709 on .Agent IdentityMCP SecurityOAuth Delegation
- Why MCP Servers Need Execution Sandboxing (And Why Your Current Stack Isn't Enough)By Om-Shree-0709 on .Agentic AiPrompt InjectionWebAssembly
MCP directory API
We provide all the information about MCP servers via our MCP API.
curl -X GET 'https://glama.ai/api/mcp/v1/servers/aws-samples/sample-dlc-mcp-server'
If you have feedback or need assistance with the MCP directory API, please join our Discord server