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 +