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

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