Skip to main content

Setup EC2 for Docker with GPU

Follow these steps to render videos on EC2 in a Docker container.
These steps are opinionated, but specify a reference that works.

Setup EC2 for Docker with GPU​

1
 Follow the instructions for GPUs on EC2. You can skip installing Chrome, Node.js and cloning the repo to render a video.
2
 Install NVIDIA Container toolkit:

Add keyring
bash
curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg
curl -s -L https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list | sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list
sudo apt-get update
Add keyring
bash
curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg
curl -s -L https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list | sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list
sudo apt-get update
Install toolkit
bash
sudo apt-get install -y nvidia-container-toolkit
Install toolkit
bash
sudo apt-get install -y nvidia-container-toolkit
3
Install Docker: 

Add Docker's official GPG key
bash
sudo apt-get update
sudo apt-get install ca-certificates curl gnupg
sudo install -m 0755 -d /etc/apt/keyrings
curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo gpg --dearmor -o /etc/apt/keyrings/docker.gpg
sudo chmod a+r /etc/apt/keyrings/docker.gpg
Add Docker's official GPG key
bash
sudo apt-get update
sudo apt-get install ca-certificates curl gnupg
sudo install -m 0755 -d /etc/apt/keyrings
curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo gpg --dearmor -o /etc/apt/keyrings/docker.gpg
sudo chmod a+r /etc/apt/keyrings/docker.gpg
Add keyring
bash
echo "deb [arch="$(dpkg --print-architecture)" signed-by=/etc/apt/keyrings/docker.gpg] https://download.docker.com/linux/ubuntu "$(. /etc/os-release && echo "$VERSION_CODENAME")" stable" | sudo tee /etc/apt/sources.list.d/docker.list > /dev/null
sudo apt-get update
Add keyring
bash
echo "deb [arch="$(dpkg --print-architecture)" signed-by=/etc/apt/keyrings/docker.gpg] https://download.docker.com/linux/ubuntu "$(. /etc/os-release && echo "$VERSION_CODENAME")" stable" | sudo tee /etc/apt/sources.list.d/docker.list > /dev/null
sudo apt-get update
Install Docker
bash
sudo apt-get install docker-ce docker-ce-cli containerd.io docker-buildx-plugin docker-compose-plugin
Install Docker
bash
sudo apt-get install docker-ce docker-ce-cli containerd.io docker-buildx-plugin docker-compose-plugin
4
Configure Docker to use the NVIDIA runtime 

Configure the NVIDIA container runtime
bash
sudo nvidia-ctk runtime configure --runtime=docker
sudo systemctl restart docker
Configure the NVIDIA container runtime
bash
sudo nvidia-ctk runtime configure --runtime=docker
sudo systemctl restart docker
5
 Create two files, Dockerfile and entrypoint.sh. You can for example create them using the nano ./file-to-create command. Use CtrlX to save and quit. 

Dockerfile
bash
FROM node:20-bookworm
RUN apt-get update
RUN apt-get install -y curl gnupg git chromium
RUN curl -fsSL https://deb.nodesource.com/gpgkey/nodesource-repo.gpg.key | gpg --dearmor -o /etc/apt/keyrings/nodesource.gpg
RUN echo "deb [signed-by=/etc/apt/keyrings/nodesource.gpg] https://deb.nodesource.com/node_18.x nodistro main" > /etc/apt/sources.list.d/nodesource.list
RUN apt-get clean
RUN rm -rf /var/lib/apt/lists/*
# Clone the repo
RUN git clone https://github.com/remotion-dev/gpu-scene.git
WORKDIR /gpu-scene
RUN npm install
# Copy the entrypoint script into the image
COPY entrypoint.sh .
CMD ["sh", "./entrypoint.sh"]
Dockerfile
bash
FROM node:20-bookworm
RUN apt-get update
RUN apt-get install -y curl gnupg git chromium
RUN curl -fsSL https://deb.nodesource.com/gpgkey/nodesource-repo.gpg.key | gpg --dearmor -o /etc/apt/keyrings/nodesource.gpg
RUN echo "deb [signed-by=/etc/apt/keyrings/nodesource.gpg] https://deb.nodesource.com/node_18.x nodistro main" > /etc/apt/sources.list.d/nodesource.list
RUN apt-get clean
RUN rm -rf /var/lib/apt/lists/*
# Clone the repo
RUN git clone https://github.com/remotion-dev/gpu-scene.git
WORKDIR /gpu-scene
RUN npm install
# Copy the entrypoint script into the image
COPY entrypoint.sh .
CMD ["sh", "./entrypoint.sh"]
entrypoint.sh
bash
#!/bin/bash
npx remotion render --gl=angle-egl Scene out/video.mp4
entrypoint.sh
bash
#!/bin/bash
npx remotion render --gl=angle-egl Scene out/video.mp4
6
 Build the container and run a sample render: 

bash
sudo docker build . -t remotion-docker-gpu
sudo docker run --gpus all --runtime=nvidia -e "NVIDIA_DRIVER_CAPABILITIES=all" remotion-docker-gpu
bash
sudo docker build . -t remotion-docker-gpu
sudo docker run --gpus all --runtime=nvidia -e "NVIDIA_DRIVER_CAPABILITIES=all" remotion-docker-gpu

See also​