Dockerfile

# attach to container in vsc, change directory to /workspace and select /opt/venv as kernel

FROM nvidia/cuda:13.2.0-cudnn-runtime-ubuntu24.04

RUN apt-get update && apt-get install -y --no-install-recommends        \
    build-essential                                                     \
    python3                                                             \
    python3-venv                                                        \
    python3-pip                                                         \
    python3-dev                                                         \
    git                                                                 \
    ninja-build                                                         \
    && rm -rf /var/lib/apt/lists/*

RUN python3 -m venv /opt/venv
ENV PATH="/opt/venv/bin:$PATH"

RUN pip install --no-cache-dir --upgrade numpy tqdm scipy matplotlib jupyterlab ipykernel
RUN pip install --no-cache-dir --upgrade torch torchvision --index-url https://download.pytorch.org/whl/cu130