See anything that can improved? Please consider creating a pull request. Thanks!

Overview

This repository showcases of how to use nix to build a VM image (with NVIDIA GPU support) for AWS deploy everything using terraform. The example in this repository

• nixos-generators is used to generate a VM image for AWS.

• NVIDIA configuration lives ./nvidia.nix

• AWS resources required to make this work

  • S3 bucket for storing the VM image
  • IAM role for the VM to access the S3 bucket
  • EBS snapshot for the VM image
  • AMI for the VM image
  • Security group for the VM
  • EC2 instance for the VM

Quick Start

Building a VM from flake.nix

nix build .#amazon --print-out-path
# Output: /nix/store/h28zy1gnyl6jmwghza1z349x0mh73w57-nixos-amazon-image-24.11.20240906.574d1ea-x86_64-linux
ls /nix/store/h28zy1gnyl6jmwghza1z349x0mh73w57-nixos-amazon-image-24.11.20240906.574d1ea-x86_64-linux
# Output: nix-support  nixos-amazon-image-24.11.20240906.574d1ea-x86_64-linux.vhd

# Modify the `vhd_image_source` in ./variables.tf variable with the path. In my case it's `/nix/store/h28zy1gnyl6jmwghza1z349x0mh73w57-nixos-amazon-image-24.11.20240906.574d1ea-x86_64-linux/nixos-amazon-image-24.11.20240906.574d1ea-x86_64-linux.vhd`

Creating AWS resources

NOTE: modify the necessary variables in variables.tf before running the following commands.

terraform apply -auto-approve
# This will take a while ~10 minutes. Most of the time is spent uploading the VHD to S3 and creating the AMI.
# To destroy the resources run: `terraform destroy -auto-approve

Contributing

Some things that this repository could use help with:

• Better documentation
• Improved terraform structure
• Home-manger for user configuration