Dockerfile to enable vulkan using Nvidia on fly.io

Dockerfile

FROM nvidia/cuda:12.2.2-base-ubuntu22.04

ENV NVIDIA_DRIVER_CAPABILITIES=compute,graphics,vulkan,utility
ENV NVIDIA_VISIBLE_DEVICES=all
ENV DEBIAN_FRONTEND=noninteractive
ENV VK_ICD_FILENAMES=/usr/share/vulkan/icd.d/nvidia_icd.json

RUN sed -i 's/ main$/ main contrib non-free/g' /etc/apt/sources.list

RUN apt-get update -y && apt-get install -y \
    curl \
    build-essential \
    ca-certificates \
    nvidia-driver-565 libnvidia-extra-565 \
    libasound2-dev \
    xorg \
    libx11-dev \
    libxkbcommon-dev \
    libnvidia-extra-565 \
    libvulkan-dev \
    vulkan-tools \
    vulkan-validationlayers \
    vulkan-validationlayers-dev \
    libvulkan1 \
    libxkbcommon0 \
    libglib2.0-0 \
    libxext6 \
    libxkbcommon0 python-is-python3 \
    pkg-config && \
    apt-get purge -y mesa-* libgl1*  && \
    # apt-get autoremove -y && \
    rm -rf /var/lib/apt/lists/*

RUN curl https://sh.rustup.rs -sSf | sh -s -- -y --default-toolchain nightly 

# ------- REPLACE THIS WITH YOUR CODE THAT DEPENDS ON VULKAN

WORKDIR /app
COPY . .

RUN /root/.cargo/bin/cargo build --release

ENTRYPOINT ["/app/target/release/xxx"]