Docker / gpuhub

Dockerfile

# Base image used as starter
FROM nvidia/cuda:12.1.1-cudnn8-runtime-ubuntu22.04

# Install necessary dependencies
RUN apt-get update && \
    apt-get upgrade -y && \
    apt-get install -y \
    python3-pip \
    git \
    ffmpeg \
    libsm6 \
    libxext6 \
    wget \
    curl \
    iproute2 \
    iputils-ping 

# Install TCMAlloc
RUN apt-get update && \
    apt-get install -y --no-install-recommends \
    libgoogle-perftools-dev \
    && apt-get autoremove -y \
    && apt-get clean \
    && rm -rf /var/lib/apt/lists/*

# Update pip
RUN pip install -U --no-cache-dir pip setuptools wheel

# Torch (2.3.0)
RUN pip install torch torchvision torchaudio

# Copy content to image
COPY ./requirements.txt /base/requirements.txt

# Install (check /requirements/pip_freeze/... if you have issues)
RUN pip install -r /base/requirements.txt

# Symlink python
RUN ln -n /usr/bin/python3 /usr/bin/python

requirements.txt

torchsde
einops
safetensors
accelerate
pyyaml
Pillow
scipy
tqdm
psutil
httpx
gradio
pytorch_lightning

xformers
diffusers
transformers
accelerate

fastapi
uvicorn[standard]

huggingface-hub[cli]

GitPython
blendmodes
clean-fid
facexlib
httpcore
inflection
jsonmerge
kornia
lark
numpy
omegaconf
open-clip-torch
piexif
resize-right
scikit-image
spandrel
tomesd
torchdiffeq
insightface
av
lxml