FROM allegroai/clearml-fractional-gpu:u22.04-cu12.3-8gb
# upgrade torch to the latest version
RUN pip3 install -U clearml pandas numpy