Docker image with coding agents pre-installed on a CUDA + PyTorch base

Melvin Vivas · X post · 2026-09-15 · Open on X

Topics: AI Dev Tools & Productivity, LLMOps, Deployment & Monitoring, Programming & ML Foundations · Level: intermediate

Summary

The creator shares his Docker image, ai-devbox. It comes with coding agents already installed on top of a CUDA and PyTorch base. It is meant for using coding agents to help with ML work on GPU clouds, and he says it works on Runpod.

Key points

• ai-devbox is a Docker image with coding agents pre-installed.
• Its base image already includes CUDA and PyTorch, so it's ready for GPU ML work.
• Use case: let coding agents help with machine-learning experiments on a GPU machine.
• The creator has tested it on Runpod GPU pods.

Resources mentioned

• ai-devbox (melvindave/ai-devbox) · tool · hub.docker.com · free
  The creator's Docker image with PyTorch, CUDA, JupyterLab and coding-agent CLIs already installed, for ML work.
  Also in: AI DevBox v1.3.0: a GPU-ready Docker image with coding-agent CLIs (Melvin Vivas on X · notes), GPU-Ready AI Devbox Docker Image on Runpod (PyTorch 2.8 + CUDA 12.8) (Melvin Vivas on X · notes)
• Runpod · tool · x.com · paid
  GPU cloud platform for running, training and serving AI models; Flash deploys workloads to it.
  Also in: AI DevBox v1.3.0: a GPU-ready Docker image with coding-agent CLIs (Melvin Vivas on X · notes), Why LLM data agents need solid data foundations (Runpod) (Melvin Vivas on X · notes), GPU devbox Docker image with coding agents on Runpod (Melvin Vivas on X · notes), GPU-Ready AI Devbox Docker Image on Runpod (PyTorch 2.8 + CUDA 12.8) (Melvin Vivas on X · notes) and 5 more
• PyTorch · tool · pytorch.org · free · recommended by both Bashiri Smith & Melvin Vivas
  Deep-learning framework. Its built-in scaled dot-product attention (SDP) was the baseline in this benchmark.
  Also in: Step-by-Step Roadmap to a $200K+ AI Engineering Role (Bashiri Smith on Facebook · notes), Wannabe vs $200K+ AI Engineer: RAG, Agents, Context and Fine-Tuning Mistakes (Bashiri Smith on Facebook · notes), AI DevBox v1.3.0: a GPU-ready Docker image with coding-agent CLIs (Melvin Vivas on X · notes), Software Engineer to AI Engineer: Job Boards, Stack, Projects and Learning Sites (Bashiri Smith on Facebook · notes) and 5 more

Try this

• Try the ai-devbox Docker image on a GPU host such as Runpod when using coding agents for ML work.