FROM allegroai/clearml-fractional-gpu:u22.04-cu12.3-8gb

# upgrade torch to the latest version
RUN pip3 install -U clearml torch torchvision torchaudio torchdata torchmetrics torchrec torchtext