Dockerfile

FROM nvidia/cuda:11.8.0-devel-ubuntu22.04

# Set environment variables to non-interactive (this prevents some prompts)
ENV DEBIAN_FRONTEND=noninteractive

# Update and install some essential packages
RUN apt-get update && apt-get install -y \
    software-properties-common \
    build-essential \
    curl \
    git \
    && rm -rf /var/lib/apt/lists/*

# install requirements for rip -r requirements.txt
RUN apt-get update && apt-get install -y \
    ffmpeg \
    libavformat-dev \
    libavcodec-dev \
    libavdevice-dev \
    libavutil-dev \
    libavutil-dev \
    libavfilter-dev \
    libswscale-dev \
    libswresample-dev

# RUN apt-get install -y libavformat-dev
# RUN apt-get install -y libavcodec-dev
# RUN apt-get install -y libavdevice-dev
# RUN apt-get install -y libavutil-dev
# RUN apt-get install -y libavfilter-dev
# RUN apt-get install -y libswscale-dev
# RUN apt-get install -y libswresample-dev

# Install Python 3 (Ubuntu 22.04 comes with Python 3.10)
RUN apt-get update && apt-get install -y python3 python3-pip

# Set the working directory inside the container
WORKDIR /usr/src/app

COPY requirements.txt ./
RUN pip3 install torch torchvision torchaudio --no-cache-dir \
    --index-url https://download.pytorch.org/whl/cu118
RUN TCNN_CUDA_ARCHITECTURES=89 pip3 install --no-cache-dir \
    git+https://github.com/NVlabs/tiny-cuda-nn/#subdirectory=bindings/torch

RUN pip3 install --no-cache-dir -r requirements.txt