FROM pytorch/pytorch:2.5.1-cuda12.4-cudnn9-runtime

COPY requirements.txt .
RUN pip install -r requirements.txt

RUN pip install jupyterlab

RUN apt update && apt install -y build-essential
