Skip to content

README.md

Latest commit

 

History

History
61 lines (40 loc) · 1.34 KB

README.md

File metadata and controls

61 lines (40 loc) · 1.34 KB

Ubuntu 22.04 NVIDIA Docker AMI Builder

This Packer configuration builds an Amazon Machine Image (AMI) based on Ubuntu 22.04 with NVIDIA drivers, CUDA toolkit, NVIDIA Container Toolkit, and Docker pre-installed.

Prerequisites

  • Packer installed
  • AWS credentials configured
  • AWS CLI installed (optional)

Features

  • Ubuntu 22.04 LTS base
  • NVIDIA Driver (535)
  • CUDA Toolkit 12.3
  • Docker CE
  • NVIDIA Container Toolkit
  • Configured for GPU workloads

Usage

  1. Initialize Packer plugins:

    packer init ubuntu-nvidia-docker.pkr.hcl

  2. Build the AMI:

    packer build ubuntu-nvidia-docker.pkr.hcl

Configuration

The build uses a g4dn.xlarge instance type by default, which includes an NVIDIA GPU. You can modify the instance_type in the configuration file if needed.

Verification

After launching an instance with this AMI, you can verify the setup:

# Check NVIDIA driver
nvidia-smi

# Check CUDA version
nvcc --version

# Test Docker with NVIDIA GPU
docker run --rm --gpus all nvidia/cuda:12.3.0-base-ubuntu22.04 nvidia-smi

Security

  • The AMI is built with the latest security updates as of build time
  • Docker is configured to use NVIDIA runtime by default
  • The ubuntu user is added to the docker group for non-root Docker access

License

MIT