From 176a748084f39839ac276936b446f870eb02041e Mon Sep 17 00:00:00 2001 From: Sebastian Raschka Date: Mon, 21 Oct 2024 18:35:04 +0000 Subject: [PATCH 1/3] Pin upper PyTorch 2.4.1 --- pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index f8b91eae2d..074c105ded 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -9,7 +9,7 @@ readme = "README.md" license = { file = "LICENSE" } dependencies = [ - "torch>=2.2.0", + "torch>=2.2.0,<=2.4.1", "numpy<2.0", "lightning==2.4.0", "jsonargparse[signatures]>=4.30.1,<=4.32.1", # 4.33 does not seem to be compatible with Python 3.9 From 8d68047adf53bf265c19ab697550b71849c25548 Mon Sep 17 00:00:00 2001 From: Sebastian Raschka Date: Mon, 21 Oct 2024 19:05:49 +0000 Subject: [PATCH 2/3] update --- .github/azure-gpu-test.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/azure-gpu-test.yml b/.github/azure-gpu-test.yml index ef50baccba..4d62f0b449 100644 --- a/.github/azure-gpu-test.yml +++ b/.github/azure-gpu-test.yml @@ -44,8 +44,8 @@ jobs: - script: | pip install --upgrade pip + pip uninstall torch # uninstall so that the latest supported torch version is installed from the .toml file in the next line pip install '.[all,test]' - pip install -U torch torchvision torchaudio displayName: 'Install dependencies' - bash: | From bd056178eaa2039e48a8a651c021b44361325c28 Mon Sep 17 00:00:00 2001 From: Sebastian Raschka Date: Mon, 21 Oct 2024 16:11:07 -0500 Subject: [PATCH 3/3] Update azure-gpu-test.yml --- .github/azure-gpu-test.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/azure-gpu-test.yml b/.github/azure-gpu-test.yml index 4d62f0b449..61b1ab96f4 100644 --- a/.github/azure-gpu-test.yml +++ b/.github/azure-gpu-test.yml @@ -44,7 +44,7 @@ jobs: - script: | pip install --upgrade pip - pip uninstall torch # uninstall so that the latest supported torch version is installed from the .toml file in the next line + pip uninstall torch -y # uninstall so that the latest supported torch version is installed from the .toml file in the next line pip install '.[all,test]' displayName: 'Install dependencies'