From a1c01438d756718ed064727b571ef1e5b64b91a9 Mon Sep 17 00:00:00 2001 From: Matteo Bunino <48362942+matbun@users.noreply.github.com> Date: Thu, 17 Aug 2023 17:24:37 +0200 Subject: [PATCH] Fix pytorch env path (#55) * Fix path * Update --------- Co-authored-by: Matteo Bunino --- use-cases/mnist/mnist-ai-train.yml | 1 + use-cases/mnist/training-workflow.yml | 6 +++--- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/use-cases/mnist/mnist-ai-train.yml b/use-cases/mnist/mnist-ai-train.yml index a2317fcc..7f423f00 100644 --- a/use-cases/mnist/mnist-ai-train.yml +++ b/use-cases/mnist/mnist-ai-train.yml @@ -11,6 +11,7 @@ train: # Lightning Trainer configuration trainer: + # Set to "cpu" when using pytorch "cpuonly" package accelerator: auto strategy: auto devices: auto diff --git a/use-cases/mnist/training-workflow.yml b/use-cases/mnist/training-workflow.yml index ca60c1e7..4ee3d220 100644 --- a/use-cases/mnist/training-workflow.yml +++ b/use-cases/mnist/training-workflow.yml @@ -20,7 +20,7 @@ steps: doc: Run MLFlow server on localhost command: python ${root}/mlflow-server.py env: - file: ${ai-root}/env-files/pytorch-lock.yml + file: ${ai-root}/env-files/pytorch-gpu-lock.yml prefix: ${ai-root}/.venv-pytorch source: ${ai-root} args: @@ -30,7 +30,7 @@ steps: doc: Train a neural network to classify MNIST images command: itwinai train env: - file: ${ai-root}/env-files/pytorch-env-gpu.yml + file: ${ai-root}/env-files/pytorch-gpu-lock.yml prefix: ${ai-root}/.venv-pytorch source: ${ai-root} args: @@ -41,7 +41,7 @@ steps: doc: Stop MLFlow server on localhost, if running command: python ${root}/mlflow-server.py env: - file: ${ai-root}/env-files/pytorch-lock.yml + file: ${ai-root}/env-files/pytorch-gpu-lock.yml prefix: ${ai-root}/.venv-pytorch source: ${ai-root} args: