From 0241c224dceb0967c60c8b2095722057c0043430 Mon Sep 17 00:00:00 2001 From: alnoam Date: Tue, 13 Feb 2024 13:34:04 +0200 Subject: [PATCH] Additional examples --- examples/{Dockerfile => Dockerfile.basic} | 2 +- examples/Dockerfile.pytorch | 5 +++++ examples/Dockerfile.tensorflow | 5 +++++ 3 files changed, 11 insertions(+), 1 deletion(-) rename examples/{Dockerfile => Dockerfile.basic} (51%) create mode 100644 examples/Dockerfile.pytorch create mode 100644 examples/Dockerfile.tensorflow diff --git a/examples/Dockerfile b/examples/Dockerfile.basic similarity index 51% rename from examples/Dockerfile rename to examples/Dockerfile.basic index d33ba98..5e917d0 100644 --- a/examples/Dockerfile +++ b/examples/Dockerfile.basic @@ -1,5 +1,5 @@ FROM allegroai/clearml-fractional-gpu:u22.04-cu12.3-8gb # upgrade torch to the latest version -RUN pip3 install -U torch torchvision torchaudio torchdata torchmetrics torchrec torchtext +RUN pip3 install -U clearml pandas numpy diff --git a/examples/Dockerfile.pytorch b/examples/Dockerfile.pytorch new file mode 100644 index 0000000..a6fbf1a --- /dev/null +++ b/examples/Dockerfile.pytorch @@ -0,0 +1,5 @@ +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 + diff --git a/examples/Dockerfile.tensorflow b/examples/Dockerfile.tensorflow new file mode 100644 index 0000000..5a3f169 --- /dev/null +++ b/examples/Dockerfile.tensorflow @@ -0,0 +1,5 @@ +FROM allegroai/clearml-fractional-gpu:u22.04-cu12.3-8gb + +# upgrade torch to the latest version +RUN pip3 install -U clearml tensorflow +