FROM pytorch/pytorch:2.1.0-cuda12.1-cudnn8-runtime
WORKDIR /workspaces/supplement

# Add utils folder to PATH
ENV PATH="/workspaces/supplement/DyGLib/utils:${PATH}"